Question

【iDesktopX】【10.2.1】模型训练报错 size mismatch for conv1.weight: copying a param with shape torch.Size([64, 3, 7, 7]) from checkpoint, the shape in current model is torch.Size([64, 4, 7, 7]).，这是什么原因？

Answer 1

【问题原因】 是因为影像是4波段8位或单波段32位真彩色的，机器学习不支持，只支持单波段24位真彩色或3波段8位的影像。 【解决办法】 需要把4波段转3波段的数据。影像文件导入为单波段的影像数据集，在数据集属性-“影像”-“波段信息”里删除第四波段，默认123对应RGB波段。得到一个三波段影像数据集即可做模型训练。