在Deepnote中运行ImageDataBunch时出现的问题

Question

我在Deepnote中运行这行代码时遇到了问题，有人知道为什么吗？

data = ImageDataBunch.from_folder(path, train="train", valid ="test",ds_tfms=get_transforms(), size=(256,256), bs=32, num_workers=4).normalize()

错误信息为：

NameError: name 'ImageDataBunch' is not defined

之前，我已经导入了Fastai库。所以我不明白！

Answer 1

Deepnote中的FastAI设置并不是那么简单。最好使用在Dockerfile中设置内容，然后一切都在笔记本中工作的custom environment。我不确定ImageDataBunch或您正在尝试做的任何事情在FastAI v1和v2中是否以相同的方式工作，但以下是v1的详细信息。

这是一个通过conda设置FastAI环境的Dockerfile文件：

# This is Dockerfile

FROM deepnote/python:3.9


RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh
RUN bash ~/miniconda.sh -b -p $HOME/miniconda
ENV PATH $HOME/miniconda/bin:$PATH
ENV PYTONPATH $HOME/miniconda

RUN $HOME/miniconda/bin/conda install python=3.9 ipykernel -y
RUN $HOME/miniconda/bin/conda install -c fastai -c pytorch fastai -y
RUN $HOME/miniconda/bin/python -m ipykernel install --user --name=conda
ENV DEFAULT_KERNEL_NAME "conda"

之后，您可以在notebook中测试fastai导入：

import fastai
from fastai.vision import *

print(fastai.__version__)

ImageDataBunch

如果您下载并解压缩this sample MNIST dataset，您应该能够按照您的建议加载数据：

data = ImageDataBunch.from_folder(path, train="train", valid ="test",ds_tfms=get_transforms(), size=(256,256), bs=32, num_workers=4).normalize()

请随时查看或克隆my Deepnote project以继续此工作。