Pytorch 是否有一种内置的方法来确保每个批次的图像大小匹配？

我有一个

torch.utills.data.Dataset

，它返回大小略有不同的图像

我使用

torch.utils.data.DataLoader

生成训练批。是否有一种内置的方法来调整图像的大小，以便它们可以在数据加载端堆叠成一批

如果没有，我可能会写一个

collate\u fn

。还是有更好的办法

---

PS：我不想将整个数据集调整为一个大小。这就是为什么我特别要求在批处理级别执行此操作。

collate\fn

是一种方法：）我同意@Proko的观点。你所要求的不是一个标准的操作，所以执行这项操作的最简单方法可能是为你的

数据加载器

定制一个

collate\fn

。是的，我对是否会有什么问题持50:50的看法。我想应该有。这似乎是一个非常明智的要求。