test.py updated
Browse files- SiamMask/tools/test.py +11 -9
SiamMask/tools/test.py
CHANGED
@@ -14,23 +14,25 @@ from os import makedirs
|
|
14 |
from os.path import join, isdir, isfile
|
15 |
import sys
|
16 |
|
17 |
-
sys.path.append(os.path.abspath(os.path.join(__file__, "..","
|
|
|
18 |
|
19 |
|
20 |
-
|
21 |
-
from
|
22 |
-
from
|
23 |
-
from
|
|
|
24 |
|
25 |
import torch
|
26 |
from torch.autograd import Variable
|
27 |
import torch.nn.functional as F
|
28 |
|
29 |
-
from
|
30 |
-
from
|
31 |
|
32 |
-
from
|
33 |
-
from
|
34 |
|
35 |
thrs = np.arange(0.3, 0.5, 0.05)
|
36 |
|
|
|
14 |
from os.path import join, isdir, isfile
|
15 |
import sys
|
16 |
|
17 |
+
sys.path.append(os.path.abspath(os.path.join(__file__, "..","..")))
|
18 |
+
sys.path.append(os.path.abspath(os.path.join(__file__, "..")))
|
19 |
|
20 |
|
21 |
+
|
22 |
+
from utils.log_helper import init_log, add_file_handler
|
23 |
+
from utils.load_helper import load_pretrain
|
24 |
+
from utils.bbox_helper import get_axis_aligned_bbox, cxy_wh_2_rect
|
25 |
+
from utils.benchmark_helper import load_dataset, dataset_zoo
|
26 |
|
27 |
import torch
|
28 |
from torch.autograd import Variable
|
29 |
import torch.nn.functional as F
|
30 |
|
31 |
+
from utils.anchors import Anchors
|
32 |
+
from utils.tracker_config import TrackerConfig
|
33 |
|
34 |
+
from utils.config_helper import load_config
|
35 |
+
from utils.pyvotkit.region import vot_overlap, vot_float2str
|
36 |
|
37 |
thrs = np.arange(0.3, 0.5, 0.05)
|
38 |
|