oguzakif commited on
Commit
1b90ade
·
1 Parent(s): 29b6148

test.py updated

Browse files
Files changed (1) hide show
  1. 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__, "..","utils")))
 
18
 
19
 
20
- from tools.utils.log_helper import init_log, add_file_handler
21
- from tools.utils.load_helper import load_pretrain
22
- from tools.utils.bbox_helper import get_axis_aligned_bbox, cxy_wh_2_rect
23
- from tools.utils.benchmark_helper import load_dataset, dataset_zoo
 
24
 
25
  import torch
26
  from torch.autograd import Variable
27
  import torch.nn.functional as F
28
 
29
- from tools.utils.anchors import Anchors
30
- from tools.utils.tracker_config import TrackerConfig
31
 
32
- from tools.utils.config_helper import load_config
33
- from tools.utils.pyvotkit.region import vot_overlap, vot_float2str
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