使用Colab从Google云端硬盘打开图片

Question

我正在使用ResNet18编码图像分类器。训练和测试阶段工作正常。 但最后我想用5-10张图像进行评估。分类器给出被分类的数据，目标和图像。提供数据和目标也很好。但是分类器从不向用户显示分类图像。

我曾经使用Show.img()在我自己的GPU上运行。

我已经将数据保存在Google云端硬盘中，并使用from colab.google import drive等装载了数据。

我需要实现哪些代码才能向用户显示图像。

感谢您的帮助。

PS：我敢打赌我错过了一个很简单的答案，但是我疯了。

Answer 1

如果您需要上传图片，可以参考此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()

使用此功能上传文件。也会保存它们。

Answer 2

像这样的东西

from IPython.display import Image

Image('drive/My Drive/folder/image.jpg')

Answer 3

首先您需要安装驱动器。打开colab，按下挂载按钮，如下图

接下来您将看到一个名为 drive/MyDrive

的新目录

您可以使用此 python 代码从中读取文件。

import cv2
import matplotlib.pyplot as plt
img = cv2.imread('drive/MyDrive/test1.jpg')
plt.imshow(img)