demo测试出错 about siamtrackers HOT 4 CLOSED

把这里的snapshot设置成绝对路径试试, 在终端cd xxx/xxx/NanoTrack 目录，然后在命令行运行脚本

from siamtrackers.

采用了你的方法，但是不行，在一开始没有动的情况下会显示RuntimeError: Error(s) in loading state_dict for ModelBuilder:
size mismatch for ban_head.corr_pw_reg.conv_kernel.0.weight: copying a param with shape torch.Size([48, 48, 1, 1]) from checkpoint, the shape in current model is torch.Size([64, 64, 1, 1]).
size mismatch for ban_head.corr_pw_reg.conv_kernel.0.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([64]).
size mismatch for ban_head.corr_pw_reg.conv_kernel.1.weight: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([64]).
size mismatch for ban_head.corr_pw_reg.conv_kernel.1.bias: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([64]).
size mismatch for ban_head.corr_pw_reg.conv_kernel.1.running_mean: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([64]).
size mismatch for ban_head.corr_pw_reg.conv_kernel.1.running_var: copying a param with shape torch.Size([48]) from checkpoint, the shape in current model is torch.Size([64]).
但是当我添加了model = nn.DataParallel(model).cuda()后而路径改为绝对路径就是报错Traceback (most recent call last):
File "demo.py", line 156, in
main()
File "demo.py", line 95, in main
model = load_pretrain(model, '/home/code/NanoTrack/models/pretrained/nanotrackv2.pth').cuda().eval() #erro
File "/home/code/NanoTrack/bin/../nanotrack/utils/model_load.py", line 70, in load_pretrain
check_keys(model, pretrained_dict)
File "/home/code/NanoTrack/bin/../nanotrack/utils/model_load.py", line 32, in check_keys
assert len(used_pretrained_keys) > 0,
AssertionError: load NONE from pretrained checkpoint
这个了，在这期间我查看了一下里面model（ModelBuilder）的信息和nanotrackv2.pth权重信息，这两个好像并不匹配，Warning: Keys in the model and checkpoint do not match.
Missing keys: {'ban_head.bbox_tower.24.running_mean', 'ban_head.bbox_tower.11.num_batches_tracked', 'ban_head.corr_pw_cls.conv.1.num_batches_tracked', 'ban_head.corr_dw_reg.conv_search.1.running_mean', 'ban_head.bbox_tower.29.bias', 'ban_head.bbox_tower.26.running_mean', 'ban_head.corr_dw_reg.conv_kernel.0.weight', 'ban_head.bbox_tower.1.running_var', 'ban_head.corr_dw_cls.conv_kernel.0.bias', 'ban_head.cls_tower.24.bias', 'ban_head.corr_pw_reg.conv.0.weight', 'ban_head.cls_tower.19.weight', 'ban_head.corr_pw_reg.conv.1.running_var', 'ban_head.bbox_tower.21.running_var', 'ban_head.corr_dw_reg.conv_kernel.1.bias', 'ban_head.cls_tower.4.weight', 'ban_head.corr_dw_reg.conv_search.1.num_batches_tracked', 'ban_head.cls_tower.4.running_var', 'ban_head.cls_tower.1.running_var', 'ban_head.bbox_tower.8.weight', 'ban_head.bbox_tower.15.weight', 'ban_head.bbox_tower.4.running_mean', 'ban_head.corr_dw_cls.conv_search.0.weight', 'ban_head.corr_pw_cls.conv.3.weight', 'ban_head.corr_pw_cls.conv.4.weight', 'ban_head.corr_pw_reg.conv.4.num_batches_tracked', 'ban_head.cls_tower.16.bias', 'ban_head.cls_tower.19.running_var', 'ban_head.corr_pw_cls.conv.1.weight', 'ban_head.cls_tower.11.num_batches_tracked', 'ban_head.cls_tower.9.weight', 'ban_head.cls_tower.24.weight', 'ban_head.cls_tower.14.running_var', 'ban_head.cls_tower.1.running_mean', 'ban_head.cls_tower.13.weight', 'ban_head.down_reg.0.weight', 'ban_head.cls_tower.24.num_batches_tracked', 'ban_head.corr_dw_reg.conv_kernel.1.running_var', 'ban_head.cls_tower.10.weight', 'ban_head.bbox_tower.14.num_batches_tracked', 'ban_head.corr_pw_reg.conv.3.weight', 'ban_head.bbox_tower.1.running_mean', 'ban_head.cls_tower.14.num_batches_tracked', 'ban_head.bbox_tower.19.running_mean', 'ban_head.cls_tower.6.weight', 'ban_head.bbox_tower.28.weight', 'ban_head.cls_tower.18.weight', 'ban_head.cls_tower.4.running_mean', 'ban_head.cls_tower.6.bias', 'ban_head.corr_pw_cls.conv.4.num_batches_tracked', 'ban_head.bbox_tower.16.num_batches_tracked', 'ban_head.bbox_tower.14.bias', 'ban_head.cls_tower.25.weight', 'ban_head.corr_dw_reg.conv_search.0.bias', 'ban_head.corr_pw_reg.conv.4.running_mean', 'ban_head.corr_pw_cls.conv.4.running_var', 'ban_head.bbox_tower.9.running_var', 'ban_head.bbox_tower.6.num_batches_tracked', 'ban_head.corr_dw_reg.conv_search.1.bias', 'ban_head.cls_tower.14.bias', 'ban_head.bbox_tower.26.running_var', 'ban_head.cls_tower.28.weight', 'ban_head.bbox_tower.24.running_var', 'ban_head.cls_tower.29.running_var', 'ban_head.cls_tower.21.running_var', 'ban_head.bbox_tower.24.num_batches_tracked', 'ban_head.corr_dw_cls.conv_search.1.weight', 'ban_head.bbox_tower.19.weight', 'ban_head.bbox_tower.24.bias', 'ban_head.cls_tower.29.bias', 'ban_head.cls_tower.4.bias', 'ban_head.corr_pw_cls.conv.4.running_mean', 'ban_head.cls_tower.3.weight', 'ban_head.bbox_tower.6.running_var', 'ban_head.cls_tower.19.bias', 'ban_head.bbox_tower.13.weight', 'ban_head.corr_pw_reg.conv.4.running_var', 'ban_head.corr_pw_reg.conv.3.bias', 'ban_head.bbox_tower.4.bias', 'ban_head.cls_tower.16.num_batches_tracked', 'ban_head.cls_tower.14.running_mean', 'ban_head.cls_tower.29.running_mean', 'ban_head.bbox_tower.21.num_batches_tracked', 'ban_head.cls_tower.26.num_batches_tracked', 'ban_head.corr_pw_reg.conv.1.running_mean', 'ban_head.cls_tower.21.running_mean', 'ban_head.bbox_tower.29.num_batches_tracked', 'ban_head.cls_tower.6.num_batches_tracked', 'ban_head.corr_dw_cls.conv_kernel.1.weight', 'ban_head.cls_tower.6.running_mean', 'ban_head.cls_tower.11.running_mean', 'ban_head.bbox_tower.26.bias', 'ban_head.cls_tower.21.bias', 'ban_head.corr_dw_cls.conv_search.0.bias', 'ban_head.bbox_tower.0.weight', 'ban_head.corr_dw_reg.conv_kernel.1.num_batches_tracked', 'ban_head.cls_tower.20.weight', 'ban_head.corr_dw_reg.conv_search.1.running_var', 'ban_head.bbox_tower.6.bias', 'ban_head.cls_tower.26.running_mean', 'ban_head.bbox_tower.6.running_mean', 'ban_head.bbox_tower.1.weight', 'ban_head.bbox_tower.3.weight', 'ban_head.corr_pw_reg.conv.1.weight', 'ban_head.bbox_tower.16.weight', 'ban_head.bbox_tower.21.running_mean', 'ban_head.bbox_tower.16.running_mean', 'ban_head.cls_tower.29.num_batches_tracked', 'ban_head.cls_tower.19.running_mean', 'ban_head.corr_dw_reg.conv_search.1.weight', 'ban_head.corr_pw_reg.conv.1.num_batches_tracked', 'ban_head.cls_tower.15.weight', 'ban_head.cls_tower.26.weight', 'ban_head.cls_tower.9.running_mean', 'ban_head.cls_tower.16.running_var', 'ban_head.corr_pw_reg.conv.1.bias', 'ban_head.down_cls.0.weight', 'ban_head.cls_tower.9.num_batches_tracked', 'ban_head.cls_tower.8.weight', 'ban_head.bbox_tower.6.weight', 'ban_head.down_cls.0.bias', 'ban_head.corr_pw_cls.conv.1.running_mean', 'ban_head.bbox_tower.11.running_mean', 'ban_head.bbox_tower.19.num_batches_tracked', 'ban_head.cls_tower.24.running_mean', 'ban_head.bbox_tower.11.weight', 'ban_head.cls_tower.29.weight', 'ban_head.bbox_tower.25.weight', 'ban_head.bbox_tower.21.weight', 'ban_head.corr_dw_cls.conv_kernel.1.running_var', 'ban_head.corr_dw_cls.conv_search.1.running_var', 'ban_head.cls_tower.11.bias', 'ban_head.bbox_tower.29.running_var', 'ban_head.cls_tower.4.num_batches_tracked', 'ban_head.corr_dw_cls.conv_kernel.0.weight', 'ban_head.corr_dw_reg.conv_search.0.weight', 'ban_head.bbox_tower.19.bias', 'ban_head.bbox_tower.26.num_batches_tracked', 'ban_head.bbox_tower.21.bias', 'ban_head.bbox_tower.9.running_mean', 'ban_head.bbox_tower.9.bias', 'ban_head.corr_pw_cls.conv.3.bias', 'ban_head.corr_pw_reg.conv.4.bias', 'ban_head.corr_dw_cls.conv_search.1.running_mean', 'ban_head.corr_dw_reg.conv_kernel.0.bias', 'ban_head.bbox_tower.16.bias', 'ban_head.bbox_tower.10.weight', 'ban_head.corr_dw_cls.conv_kernel.1.bias', 'ban_head.corr_dw_reg.conv_kernel.1.weight', 'ban_head.corr_dw_cls.conv_search.1.num_batches_tracked', 'ban_head.bbox_tower.4.weight', 'ban_head.corr_pw_cls.conv.1.running_var', 'ban_head.bbox_tower.14.weight', 'ban_head.cls_tower.6.running_var', 'ban_head.cls_tower.16.running_mean', 'ban_head.bbox_tower.4.num_batches_tracked', 'ban_head.cls_tower.1.num_batches_tracked', 'ban_head.corr_dw_reg.conv_kernel.1.running_mean', 'ban_head.cls_tower.0.weight', 'ban_head.bbox_tower.1.num_batches_tracked', 'ban_head.cls_tower.9.running_var', 'ban_head.bbox_tower.14.running_var', 'ban_head.bbox_tower.16.running_var', 'ban_head.cls_tower.21.weight', 'ban_head.cls_tower.26.running_var', 'ban_head.bbox_tower.19.running_var', 'ban_head.bbox_tower.26.weight', 'ban_head.bbox_tower.29.weight', 'ban_head.cls_tower.1.bias', 'ban_head.bbox_tower.18.weight', 'ban_head.bbox_tower.5.weight', 'ban_head.cls_logits.0.bias', 'ban_head.bbox_tower.4.running_var', 'ban_head.corr_dw_cls.conv_search.1.bias', 'ban_head.cls_tower.23.weight', 'ban_head.cls_tower.24.running_var', 'ban_head.bbox_tower.14.running_mean', 'ban_head.cls_tower.11.weight', 'ban_head.cls_tower.21.num_batches_tracked', 'ban_head.bbox_tower.9.weight', 'ban_head.cls_tower.14.weight', 'ban_head.corr_pw_cls.conv.4.bias', 'ban_head.bbox_tower.1.bias', 'ban_head.bbox_tower.11.bias', 'ban_head.corr_dw_cls.conv_kernel.1.running_mean', 'ban_head.bbox_tower.23.weight', 'ban_head.corr_dw_cls.conv_kernel.1.num_batches_tracked', 'ban_head.cls_tower.11.running_var', 'ban_head.corr_pw_cls.conv.0.weight', 'ban_head.bbox_tower.24.weight', 'ban_head.cls_tower.19.num_batches_tracked', 'ban_head.cls_logits.0.weight', 'ban_head.cls_tower.1.weight', 'ban_head.cls_tower.26.bias', 'ban_head.cls_tower.9.bias', 'ban_head.bbox_tower.29.running_mean', 'ban_head.corr_pw_reg.conv.4.weight', 'ban_head.cls_tower.5.weight', 'ban_head.bbox_tower.9.num_batches_tracked', 'ban_head.cls_tower.16.weight', 'ban_head.bbox_tower.11.running_var', 'ban_head.bbox_tower.20.weight', 'ban_head.corr_pw_cls.conv.1.bias', 'ban_head.down_reg.0.bias'}
Unexpected keys: {'ban_head.cls_pw_tower.22.running_mean', 'ban_head.bbox_pw_tower.0.weight', 'ban_head.bbox_pw_tower.2.running_mean', 'ban_head.cls_pw_tower.10.running_var', 'ban_head.cls_pw_tower.1.weight', 'ban_head.bbox_pw_tower.18.running_mean', 'ban_head.cls_pw_tower.22.running_var', 'ban_head.cls_pw_tower.18.bias', 'ban_head.cls_pw_tower.10.num_batches_tracked', 'ban_head.bbox_pw_tower.18.weight', 'ban_head.cls_pw_tower.8.weight', 'ban_head.cls_pw_tower.6.bias', 'ban_head.cls_pw_tower.10.running_mean', 'ban_head.bbox_pw_tower.4.weight', 'ban_head.bbox_pw_tower.16.weight', 'ban_head.bbox_pw_tower.18.running_var', 'ban_head.bbox_pw_tower.5.weight', 'ban_head.cls_pw_tower.14.running_var', 'ban_head.cls_pw_tower.12.weight', 'ban_head.cls_pw_tower.5.weight', 'ban_head.bbox_pw_tower.12.weight', 'ban_head.bbox_pw_tower.20.weight', 'ban_head.bbox_pw_tower.14.running_var', 'ban_head.cls_pw_tower.16.weight', 'ban_head.bbox_pw_tower.6.running_var', 'ban_head.bbox_pw_tower.2.weight', 'ban_head.bbox_pw_tower.6.num_batches_tracked', 'ban_head.bbox_pw_tower.10.num_batches_tracked', 'ban_head.bbox_pw_tower.21.weight', 'ban_head.cls_pred.0.weight', 'ban_head.bbox_pw_tower.22.running_var', 'ban_head.bbox_pw_tower.6.running_mean', 'ban_head.cls_pw_tower.2.weight', 'ban_head.cls_pw_tower.10.bias', 'ban_head.cls_pw_tower.18.running_mean', 'ban_head.cls_pw_tower.0.weight', 'ban_head.cls_pw_tower.6.weight', 'ban_head.bbox_pw_tower.22.running_mean', 'ban_head.bbox_pw_tower.18.bias', 'ban_head.cls_pw_tower.18.weight', 'ban_head.cls_pw_tower.22.num_batches_tracked', 'ban_head.cls_pw_tower.2.running_var', 'ban_head.bbox_pw_tower.9.weight', 'ban_head.cls_pw_tower.14.num_batches_tracked', 'ban_head.bbox_pw_tower.22.weight', 'ban_head.cls_pw_tower.20.weight', 'ban_head.cls_pw_tower.22.bias', 'ban_head.bbox_pw_tower.14.running_mean', 'ban_head.bbox_pw_tower.10.running_mean', 'ban_head.cls_pw_tower.2.num_batches_tracked', 'ban_head.bbox_pw_tower.13.weight', 'ban_head.cls_pw_tower.18.num_batches_tracked', 'ban_head.bbox_pw_tower.18.num_batches_tracked', 'ban_head.bbox_pw_tower.14.bias', 'ban_head.bbox_pw_tower.2.running_var', 'ban_head.bbox_pw_tower.22.bias', 'ban_head.cls_pred.0.bias', 'ban_head.bbox_pw_tower.22.num_batches_tracked', 'ban_head.cls_pw_tower.21.weight', 'ban_head.cls_pw_tower.17.weight', 'ban_head.bbox_pw_tower.6.weight', 'ban_head.cls_pw_tower.6.num_batches_tracked', 'ban_head.bbox_pw_tower.2.bias', 'ban_head.cls_pw_tower.22.weight', 'ban_head.bbox_pw_tower.8.weight', 'ban_head.bbox_pw_tower.10.running_var', 'ban_head.cls_pw_tower.14.bias', 'ban_head.bbox_pw_tower.10.weight', 'ban_head.bbox_pw_tower.6.bias', 'ban_head.bbox_pw_tower.14.weight', 'ban_head.cls_pw_tower.6.running_mean', 'ban_head.cls_pw_tower.4.weight', 'ban_head.cls_pw_tower.9.weight', 'ban_head.cls_pw_tower.13.weight', 'ban_head.cls_pw_tower.6.running_var', 'ban_head.cls_pw_tower.2.running_mean', 'ban_head.cls_pw_tower.18.running_var', 'ban_head.cls_pw_tower.14.running_mean', 'ban_head.bbox_pw_tower.14.num_batches_tracked', 'ban_head.cls_pw_tower.2.bias', 'ban_head.bbox_pw_tower.2.num_batches_tracked', 'ban_head.bbox_pw_tower.10.bias', 'ban_head.bbox_pw_tower.17.weight', 'ban_head.cls_pw_tower.14.weight', 'ban_head.bbox_pw_tower.1.weight', 'ban_head.cls_pw_tower.10.weight'}
请问这个训练的权重是不可用吗，我现在该改哪里

from siamtrackers.

from siamtrackers.

另外这里面有一些路径是需要自己修改的

按照文件步骤来，271改成511，另外建议你单步debug一下，看看出错的是在哪一行，生成空文件夹大概率就是路径不对

from siamtrackers.