hrnet-pytorch
hrnet-pytorch copied to clipboard
HRNetW48无法加载权重
使用提供的W48imagenet预训练权重,但backbone部分全部无法成功加载
pretrained = True
pretrained = True
train.py设置: backbone = "hrnetv2_w48" pretrained = True model_path = "model_data/hrnetv2_w48_imagenet_pretrained.pth"
加载失败: Load weights model_data/hrnetv2_w48_imagenet_pretrained.pth.
Successful Load Key: [] …… Successful Load Key Num: 0
Fail To Load Key: ['conv1.weight', 'bn1.weight', 'bn1.bias', 'bn1.running_mean', 'bn1.running_var', 'bn1.num_batches_tracked', 'conv2.weight', 'bn2.weight', 'bn2.bias', 'bn2.running_mean', 'bn2.running_var', 'bn2.num_batches_tracked', 'layer1.0.conv1.weight', 'layer1.0.bn1.weight', 'layer1.0.bn1.bias', 'layer1.0.bn1.running_mean', 'layer1.0.bn1.running_var', 'layer1.0.bn1.num_batches_tracked', 'layer1.0.conv2.weight', 'layer1.0.bn2.weight', 'layer1.0.bn2.bias', 'layer1.0.bn2.running_mean', 'layer1.0.bn2.running_v …… Fail To Load Key num: 1956
……好好看看注释呢
我的也是这样,解决了吗?