我加载coco数据集,然后我使用transform来调整图像大小以创建数据加载器。但我面临2个错误错误1:RuntimeError:尝试调整不可调整的存储大小错误2:RuntimeError: stack期望每个张量大小相等,但在第0项得到[3,480,640],在第1项得到[3,500,500]
我认为缩放变换不能应用于图像。
有人知道原因吗?
第一个错误表明您的数据集有一些问题。看一下您的数据集。尝试逐个加载样本,检查是否有损坏的图像。
第二个错误似乎是从pytorch数据加载器抛出的。这个错误本身是显而易见的。正如这里建议的那样,尝试这样使用transforms.Resize
:transforms.Resize((img_size, img_size))