-
Notifications
You must be signed in to change notification settings - Fork 3
/
path_config.py
113 lines (92 loc) · 3.92 KB
/
path_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
RESULTS_PATH = "tracking_results"
EVALUATION_PATH = "evaluation_results"
VISUALIZATION_PATH = "visualization_results"
# Dataset
GOT10K_PATH = "/home/heonsong/Disk2/Dataset/Got10K"
LASOT_PATH = "/home/heonsong/Disk2/Dataset/LaSOT"
NFS_PATH = "/home/heonsong/Disk2/Dataset/NFS"
OTB_PATH = "/home/heonsong/Disk2/Dataset/OTB"
OXUVA_PATH = "/home/heonsong/Disk2/Dataset/OxUvA"
TPL_PATH = "/home/heonsong/Disk2/Dataset/TColor128"
UAV_PATH = "/home/heonsong/Disk2/Dataset/UAV123"
TRACKINGNET_PATH = "/home/heonsong/Disk2/Dataset/TrackingNet"
VOT_PATH = "/home/heonsong/Disk2/Dataset/VOT2018"
OTB_NOISY_PATH = "noisy_idx"
# DaSiamRPN
DASIAMRPN_MODEL = "weights/DaSiamRPN/SiamRPNOTB.model"
# DROL
DROL_CONFIG = "weights/DROL/siamrpn_r50_l234_dwxcorr_otb/config.yaml"
DROL_SNAPSHOT = "weights/DROL/siamrpn_r50_l234_dwxcorr_otb/model.pth"
# GradNet
GRADNET_MODEL = "weights/GradNet/ckpt/base_l5_1t_49/model_epoch49.ckpt"
# MemDTC
MEMDTC_MODEL = "weights/MemDTC/models"
# MemTrack
MEMTRACK_MODEL = "weights/MemTrack/models"
# Ocean
OCEAN_MODEL = "weights/Ocean/OceanO.pth"
# RLS-RTMDNet
RLS_RTMDNET_MODEL = "weights/RLS-RTMDNet/rt-mdnet.pth"
# ROAM
ROAM_FEAT_DIR = ""
ROAM_MODEL_DIR = ""
# RPT
RPT_CONFIG = "weights/RPT/config_vot2018_offline.yaml"
RPT_SNAPSHOT = "weights/RPT/siamreppoints.model"
# SiamBAN
SIAMBAN_CONFIG = "weights/SiamBAN/siamban_r50_l234_otb/config.yaml"
SIAMBAN_SNAPSHOT = "weights/SiamBAN/siamban_r50_l234_otb/model.pth"
# SiamCAR
SIAMCAR_CONFIG = "weights/SiamCAR/siamcar_r50/config.yaml"
SIAMCAR_SNAPSHOT = "weights/SiamCAR/model_general.pth"
# SiamDW
SIAMDW_MODEL = "weights/SiamDW/CIResNet22_RPN.pth"
SIAMDW_CIRINCEP22_MODEL = "weights/SiamDW/CIRIncep22.pth"
SIAMDW_CIRNEXT22_MODEL = "weights/SiamDW/CIRNext22.pth"
SIAMDW_CIRESNET22FC_G_MODEL = "weights/SiamDW/CIResNet22FC_G.pth"
SIAMDW_CIRESNET22_MODEL = "weights/SiamDW/CIResNet22.pth"
SIAMDW_SIAMFCRES22W_MODEL = "weights/SiamDW/SiamFCRes22W.pth"
SIAMDW_CIRESNET22_RPN_MODEL = "weights/SiamDW/CIResNet22_RPN.pth"
# SiamFC
SIAMFC_MODEL = "weights/SiamFC/model.pth"
# SiamFC++
SIAMFCPP_CONFIG = "weights/SiamFC++/otb/siamfcpp_googlenet-otb.yaml"
# SiamMFC
SIAMMCF_ROOT_DIR = "external/siam-mcf/"
SIAMMCF_MODEL = "weights/SiamMCF/pretrained/siam_mcf.ckpt-50000"
# SiamRPN
SIAMRPN_MODEL = "weights/SiamRPN/model.pth"
# SiamRPN++
SIAMRPNPP_CONFIG = "weights/SiamRPN++/siamrpn_r50_l234_dwxcorr_otb/config.yaml"
SIAMRPNPP_SNAPSHOT = "weights/SiamRPN++/siamrpn_r50_l234_dwxcorr_otb/model.pth"
SIAMRPNPP_ALEXNET_OTB_CONFIG = "weights/SiamRPN++/siamrpn_alex_dwxcorr_otb/config.yaml"
SIAMRPNPP_ALEXNET_OTB_SNAPSHOT = "weights/SiamRPN++/siamrpn_alex_dwxcorr_otb/model.pth"
SIAMRPNPP_ALEXNET_CONFIG = "weights/SiamRPN++/siamrpn_alex_dwxcorr/config.yaml"
SIAMRPNPP_ALEXNET_SNAPSHOT = "weights/SiamRPN++/siamrpn_alex_dwxcorr/model.pth"
SIAMRPNPP_MOBILENET_CONFIG = (
"weights/SiamRPN++/siamrpn_mobilev2_l234_dwxcorr/config.yaml"
)
SIAMRPNPP_MOBILENET_SNAPSHOT = (
"weights/SiamRPN++/siamrpn_mobilev2_l234_dwxcorr/model.pth"
)
SIAMRPNPP_RESNET_LT_CONFIG = "weights/SiamRPN++/siamrpn_r50_l234_dwxcorr_lt/config.yaml"
SIAMRPNPP_RESNET_LT_SNAPSHOT = "weights/SiamRPN++/siamrpn_r50_l234_dwxcorr_lt/model.pth"
SIAMRPNPP_RESNET_OTB_CONFIG = (
"weights/SiamRPN++/siamrpn_r50_l234_dwxcorr_otb/config.yaml"
)
SIAMRPNPP_RESNET_OTB_SNAPSHOT = (
"weights/SiamRPN++/siamrpn_r50_l234_dwxcorr_otb/model.pth"
)
SIAMRPNPP_RESNET_CONFIG = "weights/SiamRPN++/siamrpn_r50_l234_dwxcorr/config.yaml"
SIAMRPNPP_RESNET_SNAPSHOT = "weights/SiamRPN++/siamrpn_r50_l234_dwxcorr/model.pth"
SIAMPRNPP_SIAMMASK_CONFIG = "weights/SiamRPN++/siammask_r50_l3/config.yaml"
SIAMPRNPP_SIAMMASK_SNAPSHOT = "weights/SiamRPN++/siammask_r50_l3/model.pth"
# SPM
SPM_CONFIG = "weights/SPM/alexnet_c42_otb.yaml"
# THOR
THOR_CONFIG = "weights/THOR"
THOR_SIAMFC_MODEL = "weights/THOR/SiamFC/model.pth"
THOR_SIAMRPN_MODEL = "weights/THOR/SiamRPN/SiamRPNBIG.model"
THOR_SIAMMASK_MODEL = "weights/THOR/SiamMask/model.pth"
# TRAS
TRAS_MODEL = "weights/TRAS/Student.weights"