bubbliiiing / hrnet-pytorch

这是一个hrnet-pytorch的库,可用于训练自己的语义分割数据集
MIT License
104 stars 16 forks source link

HRNetW48无法加载权重 #9

Open Cochary opened 2 years ago

Cochary commented 2 years ago

使用提供的W48imagenet预训练权重,但backbone部分全部无法成功加载

bubbliiiing commented 2 years ago

pretrained = True

Cochary commented 2 years ago

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

bubbliiiing commented 2 years ago

……好好看看注释呢

ruarua6666 commented 1 year ago

我的也是这样,解决了吗?