Skip to content

Commit 85d9ade

Browse files
lizuoxinlzx1413
lizuoxin
authored andcommitted
fix torch cudnn config & update vot test result
1 parent b0424ce commit 85d9ade

File tree

5 files changed

+16
-8
lines changed

5 files changed

+16
-8
lines changed

docs/TUTORIALS/SOT_MODEL_ZOO.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ VOT test configuration directory: _experiments/siamfcpp/test/vot_
1919
| AlexNet | SiamFCppTracker | VOT2018 |0.576 | 0.183 | 0.393| ~200| ~185 | [siamfcpp_alexnet.yaml](../../experiments/siamfcpp/test/vot/siamfcpp_alexnet.yaml)
2020
| GoogLeNet | SiamFCppTracker | VOT2018 | 0.581 | 0.169 | 0.428 | ~80 | ~65 | [siamfcpp_googlenet.yaml](../../experiments/siamfcpp/test/vot/siamfcpp_googlenet.yaml) |
2121
| GoogLeNet* | SiamFCppTracker | VOT2018 | 0.584 | 0.178 | 0.425 | ~120 | / | [siamfcpp_googlenet.yaml](../../experiments/siamfcpp/test/vot/siamfcpp_googlenet.yaml) |
22-
| AlexNet | SiamFCppOnlineTracker | VOT2018 |0.569 | 0.145 | 0.414| ~100| / | [siamfcpp_alexnet_online.yaml](../../experiments/siamfcpp/test/vot/siamfcpp_alexnet_online.yaml)
22+
| AlexNet | SiamFCppOnlineTracker | VOT2018 |0.577 | 0.136 | 0.415| ~100| / | [siamfcpp_alexnet_online.yaml](../../experiments/siamfcpp/test/vot/siamfcpp_alexnet_online.yaml)
2323

2424
\* means using TensorRT to accelerate the backbone feature extraction, and you can set **trt_mode** True in the config yaml to enable it.
2525

Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
test:
2+
track:
3+
exp_name: &TEST_NAME "siamfcpp-hpo"
4+
exp_save: &TEST_SAVE "logs/hpo"
5+
pipeline:
6+
SiamFCppOnlineTracker:
7+
test_lr_hpo_range: [0.50, 0.58] # 0.52
8+
window_influence_hpo_range: [0.20, 0.26] # 0.21
9+
penalty_k_hpo_range: [0.04, 0.08] # 0.04
10+
#online_score_weight_hpo_range: [0.4, 0.5, 0.6]

experiments/siamfcpp/test/vot/siamfcpp_alexnet_online.yaml

+4-3
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,12 @@ test:
2424
pipeline:
2525
name: "SiamFCppOnlineTracker"
2626
SiamFCppOnlineTracker:
27-
test_lr: 0.58
28-
window_influence: 0.21
29-
penalty_k: 0.08
27+
test_lr: 0.5074
28+
window_influence: 0.237
29+
penalty_k: 0.0588
3030
x_size: *X_SIZE_TEST
3131
num_conv3x3: *NUM_CONV3x3
32+
online_score_weight: 0.6
3233
tester:
3334
names: ["VOTTester",]
3435
VOTTester:

videoanalyst/engine/tester/tester_base.py

-3
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,3 @@ def set_random_seed(self):
9494
os.environ['PYTHONHASHSEED'] = str(seed)
9595
np.random.seed(seed)
9696
torch.manual_seed(seed)
97-
torch.cuda.manual_seed(seed)
98-
torch.backends.cudnn.deterministic = True
99-
torch.backends.cudnn.benchmark = True

videoanalyst/utils/hpo.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def sample_and_update_single_hp(target_cfg: CfgNode, node_name_path: List[str],
9191
random_hpo_value = np.random.randint(hpo_lb, hpo_ub)
9292
else:
9393
hpo_lb, hpo_ub = float(hpo_range[0]), float(hpo_range[1])
94-
random_hpo_value = np.random.uniform(hpo_lb, hpo_ub)
94+
random_hpo_value = round(np.random.uniform(hpo_lb, hpo_ub), 4)
9595

9696
set_cfg_value_wt_path(target_cfg, node_name_path, random_hpo_value)
9797
return random_hpo_value

0 commit comments

Comments
 (0)