Mask_RCNN:ValueError: Dimension 1 in both shapes must be equal, but are 8 and 324.

前几天有人在之前Mask_RCNN预测文章中:

https://blog.csdn.net/qq_15969343/article/details/80388311

评论了该问题:


我看了一下,主要是因为直接使用了官方给出的COCO权重进行了预测,但是可能COCO的80类物体中存在有想预测的东西,因此直接把NUM_CLASS改为了1+1;这样显然是不对的,因为1个类占了4个通道,因此是(1+80)*4=324,但是现在改为了(1+1)*4=8【8!=324】,模型自然会报错!

因此使用官方给出的权重,而不是自己训练的,同时遇到这个问题的同学,请把这个地方改为:1+80:


当然,我们可以验证一下:将1+80改为1+70,会出现以下错误:


(1+70)*4正好等于284!结论正确!

猜你喜欢

转载自blog.csdn.net/qq_15969343/article/details/80559154