Открывайте изображения с Google Drive с помощью Colab

Question

Я кодирую классификатор изображений с помощью ResNet18. Фаза Train и Test работает нормально. Но в качестве финала я хочу сделать оценку с 5-10 изображениями. Классификатор выдает данные, цель и изображения, которые классифицируются. Выдача данных и цель тоже отлично работают. Но классификатор никогда не показывает классифицированные изображения пользователю.

Запуск на моем собственном графическом процессоре, который я использовал Show.img().

Я получил свои данные на Google Диске и подключил данные с помощью from colab.google import drive и т. Д.

Какой код мне нужно реализовать, чтобы показывать изображения пользователю.

Спасибо за вашу помощь.

PS: Бьюсь об заклад, я упустил довольно простой ответ, но я схожу с ума.

Answer 1

Если вам нужно загрузить изображение, вы можете обратиться к этому ТАК :

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')