change test.py

SiamMask/tools/test.py

@@ -19,7 +19,7 @@ sys.path.append(os.path.abspath(os.path.join(__file__, "..","..","utils")))
 
 
 
-from utils.log_helper import init_log, add_file_handler
+from SiamMask.utils.log_helper import init_log, add_file_handler
 from utils.load_helper import load_pretrain
 from utils.bbox_helper import get_axis_aligned_bbox, cxy_wh_2_rect
 from utils.benchmark_helper import load_dataset, dataset_zoo

app.py

@@ -15,21 +15,21 @@ project_name = './video-object-remover'
 
 sys.path.append(project_name)
 
-sys.path.append(join(project_name, 'SiamMask',
-                'experiments', 'siammask_sharp'))
-sys.path.append(join(project_name, 'SiamMask', 'models'))
-sys.path.append(join(project_name, 'SiamMask'))
+# sys.path.append(join(project_name, 'SiamMask',
+#                 'experiments', 'siammask_sharp'))
+# sys.path.append(join(project_name, 'SiamMask', 'models'))
+# sys.path.append(join(project_name, 'SiamMask'))
 
 exp_path = join(project_name, 'SiamMask/experiments/siammask_sharp')
 pretrained_path1 = join(exp_path, 'SiamMask_DAVIS.pth')
 
-sys.path.append(join(project_name, 'FGT_codes'))
-sys.path.append(join(project_name, 'FGT_codes', 'tool'))
-sys.path.append(join(project_name, 'FGT_codes', 'LAFC', 'flowCheckPoint'))
-sys.path.append(join(project_name, 'FGT_codes', 'LAFC', 'checkpoint'))
-sys.path.append(join(project_name, 'FGT_codes', 'FGT', 'checkpoint'))
-sys.path.append(join(project_name, 'FGT_codes', 'LAFC',
-                'flowCheckPoint', 'raft-things.pth'))
+# sys.path.append(join(project_name, 'FGT_codes'))
+# sys.path.append(join(project_name, 'FGT_codes', 'tool'))
+# sys.path.append(join(project_name, 'FGT_codes', 'LAFC', 'flowCheckPoint'))
+# sys.path.append(join(project_name, 'FGT_codes', 'LAFC', 'checkpoint'))
+# sys.path.append(join(project_name, 'FGT_codes', 'FGT', 'checkpoint'))
+# sys.path.append(join(project_name, 'FGT_codes', 'LAFC',
+#                 'flowCheckPoint', 'raft-things.pth'))
 
 torch.set_grad_enabled(False)