faster r-cnn
Traceback (most recent call last):File "F:/faster-rcnn-pytorch-master/predict.py", line 14, in <module>
frcnn = FRCNN()
File "F:\faster-rcnn-pytorch-master\frcnn.py", line 88, in __init__
self.generate()
File "F:\faster-rcnn-pytorch-master\frcnn.py", line 99, in generate
self.net.load_state_dict(torch.load(self.model_path, map_location=device),False)
File "E:\anaconda3\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 1224, in load_state_dict
self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for FasterRCNN:
size mismatch for head.cls_loc.weight: copying a param with shape torch.Size() from checkpoint, the shape in current model is torch.Size().
size mismatch for head.cls_loc.bias: copying a param with shape torch.Size() from checkpoint, the shape in current model is torch.Size().
size mismatch for head.score.weight: copying a param with shape torch.Size() from checkpoint, the shape in current model is torch.Size().
size mismatch for head.score.bias: copying a param with shape torch.Size() from checkpoint, the shape in current model is torch.Size().
训练faster-r-cnn模型出现这样的问题，有大佬知道这是为什么吗？{:10_266:} :L:);P
页:
[1]