测试集评估时Shape of `pred` and `label should be equal
收藏
测试集评估时报错：Shape of `pred` and `label should be equal, but there are [1, 401, 630] and [1, 512, 512].
应该怎样解决呢？
0
收藏
请登录后评论
首先考虑训练集和测试集的数据维度是否一样，batchsize是否一样。训练的时候，数据是多大，又是如何封装，那么测试的时候数据就得多大，就得同样封装
预测结果尺寸和真值的尺寸不一致，可以检查一下推理时图像大小是否正确