Я использовал 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')