RuntimeError: The size of tensor a (786432) must match the size of tensor b (3) at non-singleton dim

RuntimeError: The size of tensor a (1572864) must match the size of tensor b (524288) at non-singleton dimension 1

在pytorch开始训练时报错,开始以为是loss函数的问题,于是将两个shape的大小强制squeeze成一样的,还是不行
最后发现问题出在输入的mask图像通道数,labelme标注转化的mask虽然是全黑的灰度图,,但是通道数居然是3!!!!

将mask通道数转为1后,正常开始训练

你可能感兴趣的:(计算机视觉,深度学习,卷积神经网络)