Browse Source

Fix: Use correct backbone weights directory for Python tracker (layer-by-layer loading)

resnet
mht 3 weeks ago
parent
commit
3e1250915e
  1. 2
      pytracking/utils/params.py

2
pytracking/utils/params.py

@ -45,7 +45,7 @@ class TrackerParams:
net = DiMPTorchScriptWrapper( net = DiMPTorchScriptWrapper(
model_dir='exported_weights', model_dir='exported_weights',
device='cuda', device='cuda',
backbone_sd='backbone_state_dict.pt',
backbone_sd='backbone',
classifier_sd='classifier_state_dict.pt', classifier_sd='classifier_state_dict.pt',
bbregressor_sd='bb_regressor_state_dict.pt' bbregressor_sd='bb_regressor_state_dict.pt'
) )

Loading…
Cancel
Save