如何将文件从Google Colab保存到Google云端硬盘

Question

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

  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中的云端硬盘读取文件非常容易，但是我找不到有关保存文件的文档。

Answer 1

首先，您必须在“编辑”模式下共享一个Google驱动器文件夹，例如： 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

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

安装驱动器

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，可以轻松找到保存在驱动器中的路径。这会将图像保存到Google云端硬盘中名为Delete的文件夹中。

# 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"复制到驱动器。