使用 numpy 数组作为 Fastai 的输入

Question

我有一个形状为 2000, 250,250,3 的 Numpy 数组

基本上是 2000 张图片，每张 250 x 250 的 RGB 波段。

我想在 Fast-AI V2 中创建一个读取 numpy 数组的数据加载器。我的怀疑是我需要将 numpy 数组转换为火炬张量。这是正确的，然后呢？

我一直在尝试查看文档，尽管它非常详尽，但我似乎无法获得所需的内容。我所能找到的只是旧版本的点点滴滴。

谢谢。