如何将文件从 Google Colab 保存到 Google Drive

Question

我正在做一些图像处理，我能够下载图像，但试图弄清楚如何将图像直接保存到谷歌驱动器。

  mask_out=cv2.subtract(rgb,img)
  mask_out=cv2.subtract(rgb,mask_out)
  mask_out[rgb == [20, 0, 25]] = 255
  cv2.imwrite('out4.jpeg' , mask_out)
  cv2_imshow(mask_out)
  cv2.waitKey(0)
  cv2.imwrite('res.png' , mask_out)    
  files.download('res.png')

您可以从代码中看到所需的图像存在于 mask-out 变量中。在 Colab 中从 Drive 读取文件很容易，但我找不到有关保存文件的文档。

Answer 1

首先你必须在“编辑”模式下共享一个谷歌驱动器文件夹，例如：https : //drive.google.com/drive/folders/1wNJUZtQD_6oBvvUhWRlAb0xjjjoLZQ?usp=sharing （链接无效，这只是一个例子)

您需要复制到代码中的 id 是： 1wNJUZtQD_6oBvvUhWRlAb0xjjjoLZQ

!pip install --upgrade gupload

from pydrive.auth import GoogleAuth
from google.colab import auth

# Authenticate and create the PyDrive client.
auth.authenticate_user()

!gupload --to '1wNJUZtQD_6oBvvUhWRlAb0xjjjoLZQ' res.png

# if multiple images
# !gupload --to '1wNJUZtQD_6oBvvUhWRlAb0xjjjoLZQ' *.png

Answer 2

安装驱动器

from google.colab import drive
drive.mount('/content/drive')

将所有内容复制到云端硬盘

!cp -r "/content/yolov5" "/content/drive/MyDrive"

Answer 3

简单的保存方法是挂载谷歌驱动器并提供驱动器中需要保存文件的文件夹的路径。

安装驱动器

from google.colab import drive
drive.mount('/content/drive')

生成随机图像并显示

import cv2
import numpy as np
from google.colab.patches import cv2_imshow

rgb_img = np.random.randint(0, 255, size=(400,400,3), dtype=np.uint8)

#cv2.imshow('RGB', rgb_img)
#cv2.waitKey(0)

cv2_imshow(rgb_img)

保存图片直接驱动

通过单击files左侧，导航到文件夹并右键单击以选择，可以轻松找到保存在驱动器中的路径Copy path。这会将图像保存到DeleteGoogle Drive 中命名的文件夹中。

# Save Image
cv2.imwrite('/content/drive/My Drive/Delete/res.png' , rgb_img)

使用复制命令

这没有被问到，但可能很有用。假设/content/res.png是图像路径，/content/drive/My Drive/Delete是目的地。

替代选项是在 colab 中写入图像并!cp "/content/res.png" "/content/drive/My Drive/Delete"在驱动器安装后使用复制到驱动器。