C-3-Framework icon indicating copy to clipboard operation
C-3-Framework copied to clipboard

RuntimeError: Error(s) in loading state_dict for CrowdCounter: Missing key(s) in state_dict: "CCN.de_pred.0.conv.bias"

Open scutzhe opened this issue 5 years ago • 5 comments

RuntimeError: Error(s) in loading state_dict for CrowdCounter: Missing key(s) in state_dict: "CCN.de_pred.0.conv.bias", "CCN.de_pred.0.conv.weight", "CCN.de_pred.1.conv.bias", "CCN.de_pred.1.conv.weight", "CCN.frontend.0.weight", "CCN.frontend.1.running_var", "CCN.frontend.1.bias", "CCN.frontend.1.weight", "CCN.frontend.1.running_mean", "CCN.frontend.4.0.conv1.weight", "CCN.frontend.4.0.bn1.running_var", "CCN.frontend.4.0.bn1.bias", "CCN.frontend.4.0.bn1.weight", "CCN.frontend.4.0.bn1.running_mean", "CCN.frontend.4.0.conv2.weight", "CCN.frontend.4.0.bn2.running_var", "CCN.frontend.4.0.bn2.bias", "CCN.frontend.4.0.bn2.weight", "CCN.frontend.4.0.bn2.running_mean", "CCN.frontend.4.0.conv3.weight", "CCN.frontend.4.0.bn3.running_var", "CCN.frontend.4.0.bn3.bias", "CCN.frontend.4.0.bn3.weight", "CCN.frontend.4.0.bn3.running_mean", "CCN.frontend.4.0.downsample.0.weight", "CCN.frontend.4.0.downsample.1.running_var", "CCN.frontend.4.0.downsample.1.bias", "CCN.frontend.4.0.downsample.1.weight", "CCN.frontend.4.0.downsample.1.running_mean", "CCN.frontend.4.1.conv1.weight", "CCN.frontend.4.1.bn1.running_var", "CCN.frontend.4.1.bn1.bias", "CCN.frontend.4.1.bn1.weight", "CCN.frontend.4.1.bn1.running_mean",

scutzhe avatar Jun 26 '19 01:06 scutzhe

解决办法 net.load_state_dict(torch.load(model_path),strict=False) 加上 strict=False

scutzhe avatar Jun 26 '19 03:06 scutzhe

Thank you:)

Tu-Jiawei avatar Jun 27 '19 01:06 Tu-Jiawei

使用这种方法,最终预测结果差很多。

PuddingCoder avatar Oct 09 '19 06:10 PuddingCoder

Same here, with strict=False I got people count always 0, and quite strange density maps. Here is the result for ShanghaiTech Part_B IMG_153.jpg with 11-ResSFCN-101_all_ep_94_mae_26.8_mse_66.1.pth pretrained model.

dn_IMG_153

Artyrm avatar Aug 19 '20 14:08 Artyrm

@all Please visit this issue: https://github.com/gjy3035/C-3-Framework/issues/22#issuecomment-494287652

gjy3035 avatar Aug 20 '20 00:08 gjy3035