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冲突，改一下制作标签的代码就好。