torch.onnx.errors.OnnxExporterError: Module onnx is not installed!

安装python包 onnx

pip install onnx

ValueError: operands could not be broadcast together with shapes。。。

我是在训练分割模型,出现这个问题是矩阵相乘时不匹配宽高,发生冲突。

我检查了我的标签图片,0为背景,1为分割类别,却出现了2的值,所以2和3冲突,改一下制作标签的代码就好。

更多推荐