Faster-rcnn:RuntimeError: Error(s) in loading state_dict for FasterRCNN
在进行预测的时候报错,报错如下:
该问题主要是由于训练时的标签类别和预测时的标签类别不一致所导致的,将predict.py里的
num_classes修改和训练时的保持一致即可。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
二维码