change test.py
Browse files- SiamMask/tools/test.py +7 -7
SiamMask/tools/test.py
CHANGED
@@ -20,19 +20,19 @@ sys.path.append(os.path.abspath(os.path.join(__file__, "..","..","utils")))
|
|
20 |
|
21 |
|
22 |
from SiamMask.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 |
|
|
|
20 |
|
21 |
|
22 |
from SiamMask.utils.log_helper import init_log, add_file_handler
|
23 |
+
from SiamMask.utils.load_helper import load_pretrain
|
24 |
+
from SiamMask.utils.bbox_helper import get_axis_aligned_bbox, cxy_wh_2_rect
|
25 |
+
from SiamMask.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 SiamMask.utils.anchors import Anchors
|
32 |
+
from SiamMask.utils.tracker_config import TrackerConfig
|
33 |
|
34 |
+
from SiamMask.utils.config_helper import load_config
|
35 |
+
from SiamMask.utils.pyvotkit.region import vot_overlap, vot_float2str
|
36 |
|
37 |
thrs = np.arange(0.3, 0.5, 0.05)
|
38 |
|