python - fast.ai 名称 ImageDataBunch 未找到

Question

我正在研究图像识别系统，但在 fast.ai 加载数据时遇到了问题。

我的代码是：

from fastai.vision import *
import numpy as np


np.random.seed(42)

data = ImageDataBunch.from_folder('D:\\Apple-detector\\data\\fruits', train=".", valid_pct=0.2, ds_tfms=get_transforms(), size=450, num_workers=4, bs = 16).normalize(imagenet_stats)

print(data.classes())

它返回错误： NameError: name 'ImageDataBunch' is not defined

标签： pythoncomputer-visionobject-detectionfast-ai