Comments (4)
把这里的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.
Related Issues (20)
- 有没有裁剪Lasot的代码,博主? HOT 1
- Pretrained model of Updatenet Resnet256 HOT 1
- 关于Siamese tracker的搜参 HOT 1
- nanotrackv3.pth HOT 1
- 博主您好,请问我自己转出来的nanotrackv1版本的onnx文件backbone部分的大小900多kb可能是什么原因呢 HOT 1
- Lighttrack代码 HOT 1
- COCO百度云链接失效 HOT 1
- 求TrTr预训练模型,给的下载链接打不开了 HOT 1
- How the nanotrack onnx files works without opencv HOT 1
- Nanotrack应用到了lighttrack那里?有应用到一次性NAS么? HOT 1
- 可否提供更多实验的backbone的预训练权重? HOT 3
- 重新训练NanoTrackV3时,报错:NaN or Inf found in input tensor. HOT 6
- baidu Datasets Access HOT 1
- ValueError HOT 10
- AttributeError: 'list' object has no attribute 'items' HOT 1
- nanotrackv3 转 ncnn模型,pth转pt,head的推理结果异常nan HOT 11
- Training regiment? HOT 4
- NanoTrack训练代码问题
- NanoTrack模型代码中没有找到使用颈部网络的地方 HOT 2
- nanotrack HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from siamtrackers.