-
Notifications
You must be signed in to change notification settings - Fork 0
/
test.py
33 lines (24 loc) · 1.03 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from __future__ import absolute_import
from got10k.experiments import *
from siamfc import TrackerSiamFC
from config import config
import os
import glob
import numpy as np
if __name__ == '__main__':
# setup the tracker to access the pre-trained model
folder_path = 'model'
results = 'results'
reports = 'reports'
model=np.sort(glob.glob(os.path.join(folder_path,"*.pth")))
for i in model:
model_name = os.path.splitext(os.path.basename(i))[0]
results_path_bbox = os.path.join(results,model_name)
reports_path_graph = os.path.join(reports,model_name)
tracker_test = TrackerSiamFC(net_path=i)
experiments = ExperimentOTB(config.OTB_dataset_directoty, version=2015,
result_dir=results_path_bbox,
report_dir=reports_path_graph)
# run the experiments for tracking to report the performance
experiments.run(tracker_test, visualize=False)
experiments.report([tracker_test.name])