python - PyTorch 数据集中哪里使用了 len 函数？

Question

我希望使用 here 中的代码。但是，我正在查看框 5，其中有以下函数；

def __len__(self):
    # Default epoch size is 10 000 samples
    return 10000

我在整个脚本中没有看到任何使用此函数的地方。如果能对此进行澄清，我们将不胜感激。

此外，我想确定用于训练该卷积神经网络的图像 block 的数量。这个 len 函数与补丁的数量有关吗？

Answer 1

这是 Dataset 的函数类(class)。 __len__() 函数指定数据集的大小。在您引用的代码中，在框 10 中，数据集被初始化并传递到 DataLoader对象:

train_set = ISPRS_dataset(train_ids, cache=CACHE)
train_loader = torch.utils.data.DataLoader(train_set,batch_size=BATCH_SIZE)

您会看到在 DataLoader 中传递了数据集对象以及批量大小。然后，DataLoader 对象使用数据集的 __len__ 函数来创建批处理。这发生在框 13 中，它在 DataLoader 上进行迭代。