Faster-rcnn:RuntimeError: Error(s) in loading state_dict for FasterRCNN

在进行预测的时候报错,报错如下: 

 该问题主要是由于训练时的标签类别和预测时的标签类别不一致所导致的,将predict.py里的

num_classes修改和训练时的保持一致即可。

 

你可能感兴趣的:(python)