search

rafellerc / Pytorch-SiamFC

Pytorch implementation of "Fully-Convolutional Siamese Networks for Object Tracking"
Other
608 stars 148 forks source link

can't find "baseline-conv5-original" network #5

Open arbitularov opened 5 years ago

arbitularov commented 5 years ago

Here "Download the state dictionary of the Network and put it inside of Root_Dir/models and set the value of net in Root_Dir/tracking/experiments//design.json as the name of the network (e.g. baseline-conv5-original)" I don't understand how to take " baseline-conv5-original" network. I did it: "net": "baseline_pretrained.pth.tar" Then I have a mistake: xgbj@Name:/Tracker/Pytorch-SiamFC$ chmod -x run_tracker_evaluation.py xgbj@Name:/Tracker/Pytorch-SiamFC$ ./run_tracker_evaluation.py Model set to GPU

Traceback (most recent call last): File "./run_tracker_evaluation.py", line 235, in main() File "./run_tracker_evaluation.py", line 32, in main tracking_graph = TrackingGraph(design, env, hp) File "/Tracker/Pytorch-SiamFC/tracking/tracking_graph.py", line 43, in init self.siamese = load_model(self.siamese, design.net) File "/Tracker/Pytorch-SiamFC/tracking/networks.py", line 90, in load_model model.load_state_dict(torch.load(model_path)) File "/home/xgbj/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 719, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for SiamFC: Missing key(s) in state_dict: "branch.0.weight", "branch.0.bias", "branch.1.weight", "branch.1.bias", "branch.1.running_mean", "branch.1.running_var", "branch.4.weight", "branch.4.bias", "branch.5.weight", "branch.5.bias", "branch.5.running_mean", "branch.5.runn ing_var", "branch.8.weight", "branch.8.bias", "branch.9.weight", "branch.9.bias", "branch.9.running_mean", "branch.9.running_var", "bra nch.11.weight", "branch.11.bias", "branch.12.weight", "branch.12.bias", "branch.12.running_mean", "branch.12.running_var", "branch.14.w eight", "branch.14.bias", "match_batchnorm.weight", "match_batchnorm.bias", "match_batchnorm.running_mean", "match_batchnorm.running_va r". Unexpected key(s) in state_dict: "epoch", "state_dict", "optim_dict".

rafellerc commented 5 years ago

You're right, there's a conflict between the Training and Tracking code. Somewhere along the development the definitions diverged. I'll be working to correct it and get back to you ASAP.

arbitularov commented 5 years ago

Thank you! I will be waiting for your answer.