不同大小的图像coco数据集



我加载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))

相关内容

  • 没有找到相关文章

最新更新