-
Notifications
You must be signed in to change notification settings - Fork 34
/
Copy pathconfig.yml
executable file
·42 lines (39 loc) · 2.05 KB
/
config.yml
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
INTERACTIVE_MODELS_PATH: "./weights"
EXPS_PATH: "/playpen-raid2/qinliu/models/model_0125_2023"
# Evaluation datasets
GRABCUT_PATH: "/playpen-raid2/qinliu/data/GrabCut"
BERKELEY_PATH: "/playpen-raid2/qinliu/data/Berkeley"
DAVIS_PATH: "/playpen-raid2/qinliu/data/DAVIS345"
COCO_MVAL_PATH: "/playpen-raid2/qinliu/data/COCO_MVal"
BraTS_PATH: "/playpen-raid2/qinliu/data/BraTS20"
ssTEM_PATH: "/playpen-raid2/qinliu/data/ssTEM"
OAIZIB_PATH: "/playpen-raid2/qinliu/data/OAI-ZIB/iseg_slices"
OAI_PATH: "/playpen-raid2/qinliu/data/OAI"
HARD_PATH: "/playpen-raid2/qinliu/data/HARD"
# Train datasets
SBD_PATH: "/playpen-raid2/qinliu/data/SBD/dataset"
COCO_PATH: "/playpen-raid2/qinliu/data/COCO_2017"
LVIS_v1_PATH: "/playpen-raid2/qinliu/data/COCO_2017"
OPENIMAGES_PATH: "./datasets/OpenImages"
PASCALVOC_PATH: "/playpen-raid2/qinliu/data/PascalVOC"
ADE20K_PATH: "./datasets/ADE20K"
# You can download the weights for HRNet from the repository:
# https://github.com/HRNet/HRNet-Image-Classification
IMAGENET_PRETRAINED_MODELS:
HRNETV2_W18: "./weights/pretrained/hrnetv2_w18_imagenet_pretrained.pth"
HRNETV2_W32: "./weights/pretrained/hrnetv2_w32_imagenet_pretrained.pth"
HRNETV2_W40: "./weights/pretrained/hrnetv2_w40_imagenet_pretrained.pth"
HRNETV2_W48: "./weights/pretrained/hrnetv2_w48_imagenet_pretrained.pth"
HRNETV2_W64: "./weights/pretrained/hrnetv2_w64_imagenet_pretrained.pth"
MIT_B5: "./weights/pretrained/mit_b5_converted.pth"
MIT_B4: "./weights/pretrained/mit_b4_converted.pth"
MIT_B3: "./weights/pretrained/mit_b3_converted.pth"
MIT_B2: "./weights/pretrained/mit_b2_converted.pth"
MIT_B1: "./weights/pretrained/mit_b1_converted.pth"
MIT_B0: "./weights/pretrained/mit_b0_converted.pth"
HRF_BASE: "./weights/pretrained/hrt_base.pth"
SWIN_BASE: "./weights/pretrained/swin_base_patch4_window12_384_22k.pth"
SWIN_LARGE: "./weights/pretrained/swin_large_patch4_window12_384_22k.pth"
MAE_BASE: "./weights/pretrained/mae_pretrain_vit_base.pth"
MAE_LARGE: "./weights/pretrained/mae_pretrain_vit_large.pth"
MAE_HUGE: "./weights/pretrained/mae_pretrain_vit_huge.pth"