【已解决】RuntimeError: stack expects each tensor to be equal size, but got [1, 640, 640] at entry 0...

【报错】

return torch.stack(batch, 0, out=out)

^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

RuntimeError: stack expects each tensor to be equal size, but got [1, 640, 640] at entry 0 and [1, 224, 640, 640] at entry 23

【分析】

根据网上其他方案检查transforms.Resize()参数设置，未出现transforms.Resize((640,640))写为transforms.Resize((640))的问题，进一步检查，发现不匹配的tensor形状 [1, 224, 640, 640]中出现了原图尺寸224，可能是输入图像维度错位导致

【原因】

数据集中不全为单通道图像，其中掺杂了RGB等图像

【解决】

读取到image后，加上：

if len(image.shape) > 2:
image = image[:, :, 0]