clean up sot part

eval.sh

@@ -10,9 +10,9 @@ EXP_NAME=$1
 CFG_PATH=config/${EXP_NAME}.yaml
 SMRY_ROOT=results/summary/${EXP_NAME}
 mkdir -p $SMRY_ROOT
-#CUDA_VISIBLE_DEVICES=$2 python -u test/test_sot_siamfc.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/sot_siamfc.log 2>&1 
-#CUDA_VISIBLE_DEVICES=$2 python -u test/test_sot_cfnet.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/sot_cfnet.log 2>&1
+CUDA_VISIBLE_DEVICES=$2 python -u test/test_sot_siamfc.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/sot_siamfc.log 2>&1 
+CUDA_VISIBLE_DEVICES=$2 python -u test/test_sot_cfnet.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/sot_cfnet.log 2>&1
 CUDA_VISIBLE_DEVICES=$2 python -u test/test_vos.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/vos.log 2>&1
-#CUDA_VISIBLE_DEVICES=$2 python -u test/test_mot.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/mot.log 2>&1
-#CUDA_VISIBLE_DEVICES=$2 python -u test/test_mots.py --config $CFG_PATH | tee  results/summary/${EXP_NAME}/mots.log 2>&1
-#CUDA_VISIBLE_DEVICES=$2 python -u test/test_posetrack.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/posetrack.log 2>&1
+CUDA_VISIBLE_DEVICES=$2 python -u test/test_mot.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/mot.log 2>&1
+CUDA_VISIBLE_DEVICES=$2 python -u test/test_mots.py --config $CFG_PATH | tee  results/summary/${EXP_NAME}/mots.log 2>&1
+CUDA_VISIBLE_DEVICES=$2 python -u test/test_posetrack.py --config $CFG_PATH | tee results/summary/${EXP_NAME}/posetrack.log 2>&1

requirements.txt

@@ -10,3 +10,4 @@ pycocotools==2.0
 lap==0.4.0
 click==7.1.2
 motmetrics==1.2.0
+scikit-image==0.16.2

test/test_sot_cfnet.py

@@ -18,7 +18,7 @@
 import torch
 import numpy as np
 
-import tracker.sot.lib.models.models as models
+import tracker.sot.lib.models as models
 from tracker.sot.lib.utils.utils import  load_dataset, crop_chw, \
     gaussian_shaped_labels, cxy_wh_2_rect1, rect1_2_cxy_wh, cxy_wh_2_bbox
 from tracker.sot.lib.core.eval_otb import eval_auc_tune

test/test_sot_siamfc.py

@@ -18,7 +18,7 @@
 import torch
 import numpy as np
 
-import tracker.sot.lib.models.models as models
+import tracker.sot.lib.models as models
 from tracker.sot.lib.utils.utils import  load_dataset, crop_chw, \
     gaussian_shaped_labels, cxy_wh_2_rect1, rect1_2_cxy_wh, cxy_wh_2_bbox
 from tracker.sot.lib.core.eval_otb import eval_auc_tune

test/test_vos.py

@@ -42,6 +42,7 @@ def main(args, vis):
         test_loss = test(val_loader, model, args)
 
     cvt_path = args.save_path.replace(args.exp_name, 'convert_'+args.exp_name)
+
     # convert to DAVIS format
     cvt_cmd_str = ('python ./eval/convert_davis.py --in_folder {} --out_folder {} '
                '--dataset {}').format(args.save_path, cvt_path, args.davisroot)

tracker/sot/lib/models/__init__.py

@@ -0,0 +1,6 @@
+from .cfnet import CFNet
+from .siamfc import SiamFC
+from .connect import box_tower, AdjustLayer, AlignHead, Corr_Up, MultiDiCorr, OceanCorr
+from .backbones import ResNet50, ResNet22W
+from .modules import MultiFeatureBase
+