使用pytorch从colab保存到谷歌驱动器会产生黑色图像答案

【问题标题】：Saving from colab to google drive with pytorch makes black image使用pytorch从colab保存到谷歌驱动器会产生黑色图像
【发布时间】：2021-03-19 07:41:31
【问题描述】：

我用 Colab 安装了一个谷歌驱动器：

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

这很好用。使用 pytorch 将张量图像以 png 格式保存到驱动器：

robin = image.type(torch.uint8)
torchvision.io.write_png(robin, '/content/drive/My Drive/ImgFolder/robin.png')

在驱动器中保存一个名为 robin.png 的文件，该文件只是一个黑色方块。使用 plot.imshow 查看张量图像，我可以看到它是知更鸟的图像，但由于某种原因将其保存到驱动器使其变黑。图片来自 STL10 数据集。

这里是重现问题的笔记本的链接：https://colab.research.google.com/drive/1ayAs7HmzuA-eRBnat9HoziMM4yDdlyTA?usp=sharing

【问题讨论】：

请分享一个能够重现您观察到的问题的独立笔记本。
我刚刚添加了一个

标签： google-drive-api pytorch png google-colaboratory

【解决方案1】：

我只使用 PLT 库进行图像读取和进一步处理。

from PIL import Image
import os
from pathlib import Path


path = r"/content/drive/MyDrive/aa/train"
os.chdir(path)

file_path = f"/content/drive/MyDrive/aa/train/xyz.jpeg"  # Image file path

## image cropping and saving 

img = Image.open(file_path)
img = img.resize((250,250), Image.ANTIALIAS)
img.save("/content/drive/MyDrive/aa/xyz.jpeg") # Image saving to another directory

因此您可以轻松地从谷歌驱动器读取图像并将图像从谷歌 colab 保存到谷歌驱动器。

【讨论】：