pytorch 深度学习,transforms.Resize统一图片大小之后,运行一会还是出现维度大小不统一的问题

在做pytorch深度学习了时,已经通过transforms.Compose([transforms.ToTensor(), transforms.Resize([300, 300])]),将收集的图片拉伸缩放到相同的大小,但还是报错,而且是正常运行一会之后突然报错

img

img

报错给出了 mat1 和 mat2 的形状是不同的,你就去检查一下这两个数组的形状啊。如果确实不同(错误提示中已经有了),再查为什么不同。

程序也没有贴全,报错的 90行 114行都没有

错误可能是由于图片大小不一致造成的,建议检查一下你的数据集中是否有尺寸不一致的图片,或者在数据预处理的时候添加一些针对不一致图片的处理方式,例如跳过这些图片。同时检查一下网络的输入大小是否和你预处理的大小相同。