Как предсказать изображения с помощью FASTI - PullRequest
0 голосов
/ 12 апреля 2020

Я использовал fastai для создания модели классификации изображений. Я сохранил модель, используя learn.save('CNN-B30',return_path=True). Я хочу сделать прогнозирование, используя эту модель, моя тестовая папка содержит несколько изображений для прогнозирования. как получить доступ к изображениям тестовой папки и l oop всем изображениям для прогнозирования.

Моя папка набора данных:
набор данных / поезд / категория1, категория2, категория3
набор данных / действительный / категория1, категория2, category3
набор данных / test / image1.jpg, image2.jpg

Это код, который я использовал для загрузки модели

%reload_ext autoreload
%autoreload 2
%matplotlib inline
from fastai.vision import *
from fastai.metrics import error_rate

path = Path(r'C:\Users\Gayathra\.fastai\data\mnist_tiny')

learn = load_learner(path)
learn = load_learner(path, 'CNN-B30-model')
...