|
@ -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' |
|
|
) |
|
|
) |
|
|