【问题标题】:Open Images from Google Drive using Colab使用 Colab 从 Google Drive 打开图像
【发布时间】:2019-10-28 05:30:56
【问题描述】:
我正在使用 ResNet18 编写图像分类器。训练和测试阶段工作正常。
但作为决赛,我想用 5-10 张图像进行评估。分类器给出数据、目标和分类的图像。提供数据和目标也可以正常工作。但分类器从不向用户显示分类图像。
在我自己的 GPU 上运行,我曾经使用 Show.img()。
我已经在我的 Google Drive 上获取了我的数据,并使用 from colab.google import drive 等挂载了这些数据。
我需要实现哪些代码来向用户显示图像。
感谢您的帮助。
PS:我敢打赌我错过了一个非常简单的答案,但我快疯了。
【问题讨论】:
标签:
python-3.x
google-drive-api
google-colaboratory
【解决方案1】:
首先您需要安装驱动器。打开colab,按下mount按钮,如下图
接下来你会看到一个名为drive/MyDrive的新目录
您可以使用此 python 代码从中读取文件。
import cv2
import matplotlib.pyplot as plt
img = cv2.imread('drive/MyDrive/test1.jpg')
plt.imshow(img)
【解决方案2】:
像这样的
from IPython.display import Image
Image('drive/My Drive/folder/image.jpg')
【解决方案3】:
如果需要上传图片,可以参考这个SO post:
import cv2
import numpy as np
from matplotlib import pyplot as plt
img = cv2.imread("Sample-image.jpg")
img_cvt=cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.imshow(img_cvt)
plt.show()
使用此功能上传文件。它也会保存它们。