bubbliiiing / faster-rcnn-pytorch

这是一个faster-rcnn的pytorch实现的库,可以利用voc数据集格式的数据进行训练。
MIT License
1.6k stars 357 forks source link

无法正确装载模型数据 #98

Open bettermorn opened 2 years ago

bettermorn commented 2 years ago

谢谢Up主的劳动!请教下,除了voc_weights_resnet.pth模型,以下模型的数据无法正确装载,请问能否更新下? 1.resnet50-19c8e357.pth 2.vgg16-397923af.pth 错误信息如下: RuntimeError: Error(s) in loading state_dict for FasterRCNN: Missing key(s) in state_dict: "extractor.0.weight", "extractor.1.weight", "extractor.1.bias", "extractor.1.running_mean", "extractor.1.running_var", 以下是两个成功模型的使用方法: 模型voc_weights_vgg.pth 对应bakcbone = vgg
模型voc_weights_resnet.pth 对应 backbone = resnet60 非常感谢!

bubbliiiing commented 2 years ago

pretrain = True