add detector

.gitignore

@@ -1,5 +1,4 @@
 docs/test_video.mp4
-detector/
 test/test_tao.py
 config/tao*
 tracker/mot/tao.py

config/got10k_i18s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: got10k_imagenet_resnet18_s3
+
+    # Model related
+    model_type: imagenet18
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'GOT10KVAL'
+    dataroot: '/home/wangzd/datasets/GOT/GOT10k/'
+    epoch_test: False
+

config/got10k_i18s3_test.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: got10k_imagenet_resnet18_s3_test
+
+    # Model related
+    model_type: imagenet18
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'GOT10K'
+    dataroot: '/home/wangzd/datasets/GOT/GOT10k/'
+    epoch_test: False
+

config/got10k_i50s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: got10k_imagenet_resnet50_s3
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'GOT10KVAL'
+    dataroot: '/home/wangzd/datasets/GOT/GOT10k/'
+    epoch_test: False
+

config/got10k_i50s3_test.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: got10k_imagenet_resnet50_s3_test
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'GOT10K'
+    dataroot: '/home/wangzd/datasets/GOT/GOT10k/'
+    epoch_test: False
+

config/lasot_i50s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: lasot_imagenet_resnet50_s3
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'LASOT'
+    dataroot: '/home/wangzd/datasets/GOT/LaSOT/'
+    epoch_test: False
+

config/lasot_mocov1s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: lasot_mocov1_s3
+
+    # Model related
+    model_type: moco-v1 
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/moco-v1.pth' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [4,5,6,7]
+    device: cuda
+
+sot:
+    dataset: 'LaSOT'
+    dataroot: '/home/wangzd/datasets/GOT/LaSOT/'
+    epoch_test: False
+

config/lasot_pclv2s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: lasot_pclv2_s3
+
+    # Model related
+    model_type: pcl-v2 
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/pcl-v2.pth' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'LaSOT'
+    dataroot: '/home/wangzd/datasets/GOT/LaSOT/'
+    epoch_test: False
+

config/lasot_selav2s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: lasot_selav2_s3
+
+    # Model related
+    model_type: sela-v2 
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/sela-v2.pth' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [4,5,6,7]
+    device: cuda
+
+sot:
+    dataset: 'LaSOT'
+    dataroot: '/home/wangzd/datasets/GOT/LaSOT/'
+    epoch_test: False
+

config/lasot_vfss3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: lasot_vfs_s3
+
+    # Model related
+    model_type: vfs
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/vfs/otb1.pth.tar' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'LaSOT'
+    dataroot: '/home/wangzd/datasets/GOT/LaSOT/'
+    epoch_test: False
+

config/tc128_i50s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: tc128_imagenet_resnet50_s3
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'TC128'
+    dataroot: '/home/wangzd/datasets/GOT/TC128/temple-color-128/'
+    epoch_test: False
+

config/tlp_i50s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: tlp_imagenet_resnet50_s3
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'TLP'
+    dataroot: '/home/wangzd/datasets/GOT/TLP/TLP/'
+    epoch_test: False
+

config/tlp_pclv2s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: tlp_pclv2_resnet50_s3
+
+    # Model related
+    model_type: pcl-v2 
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/pcl-v2.pth' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'TLP'
+    dataroot: '/home/wangzd/datasets/GOT/TLP/TLP/'
+    epoch_test: False
+

config/tlp_selav2s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: tlp_selav2_resnet50_s3
+
+    # Model related
+    model_type: sela-v2
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/sela-v2.pth' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'TLP'
+    dataroot: '/home/wangzd/datasets/GOT/TLP/TLP/'
+    epoch_test: False
+

config/tlp_vfsv3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: tlp_vfs_resnet50_s3
+
+    # Model related
+    model_type: vfs
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: './weights/vfs/otb1.pth.tar' 
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [4,5,6,7]
+    device: cuda
+
+sot:
+    dataset: 'TLP'
+    dataroot: '/home/wangzd/datasets/GOT/TLP/TLP/'
+    epoch_test: False
+

config/trackingnet_i50s3.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: trackingnet_imagenet_resnet50_s3
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'TrackingNetVAL'
+    dataroot: '/home/wangzd/datasets/GOT/TrackingNet/'
+    epoch_test: False
+

config/trackingnet_i50s3_test.yaml

@@ -0,0 +1,23 @@
+common:
+    exp_name: trackingnet_imagenet_resnet50_s3_test
+
+    # Model related
+    model_type: imagenet50
+    remove_layers: ['layer4']
+    im_mean: [0.485, 0.456, 0.406]
+    im_std: [0.229, 0.224, 0.225]
+    nopadding: False
+    resume: None
+
+    # Misc
+    down_factor: 8
+    infer2D: True 
+    workers: 4
+    gpu_id: [0,1,2,3]
+    device: cuda
+
+sot:
+    dataset: 'TrackingNet'
+    dataroot: '/nas/zhangjw/TrackingNet/'
+    epoch_test: False
+