【报错】
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]