python - 名称“ImageDataBunch”未定义

Question

我收到此错误“名称 'ImageDataBunch' 未定义”并且无法在 Internet 上找到解决方案。

完整的代码可以在这里找到：https ://dzlab.github.io/jekyll/update/2018/11/13/audio-classification/ 。

np.random.seed(42)
data = ImageDataBunch.from_lists(path, fnames, labels, ds_tfms=None, size=224, bs=bs)
data.normalize(imagenet_stats)
data.show_batch(rows=5, figsize=(8,8))

我在 Google Colab 和 Python 3.6 上运行此代码。

标签： pythongoogle-colaboratory

Answer 1

您只需ImageDataBunch要从 fast.ai 库中导入该类。这是它的文档。只需将以下内容添加到代码顶部即可导入整个视觉库类：

from fastai.vision import *