如何找到单个图像实例的分类？ 法式

问题描述

下面的代码从MNIST数据集中为3和7个图像数字创建一个分类器。 您能帮我弄清楚如何使用此学习器对新图像进行分类吗？

!pip install -Uqq fastbook
import fastbook
fastbook.setup_book()

from fastai.vision.all import *
from fastbook import *

matplotlib.rc('image',cmap='Greys')
path = untar_data(URLs.MNIST_SAMPLE)
dls = ImageDataLoaders.from_folder(path)
learn=cnn_learner(dls,resnet18,pretrained=False,loss_func=F.cross_entropy,metrics = accuracy)
learn.fit_one_cycle(1,0.1)

解决方法

只需使用learn.predict('testImage.jpg')

供参考，请参阅fastai文档vision tutorial