numpy - 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 设置并不是那么简单。最好使用 custom environment您在 Dockerfile 中设置内容的地方然后一切都在笔记本中工作。我不确定 ImageDataBunch或者您尝试做的任何事情在 FastAI v1 和 v2 中都以相同的方式工作，但这里是 v1 的详细信息。
这是一个 Dockerfile，它通过 conda 设置 FastAI 环境:

# 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"

之后，您可以测试 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继续努力。