-
Notifications
You must be signed in to change notification settings - Fork 5
/
PanopticFCN-R50+FaPN-1x-FAST.yaml
48 lines (46 loc) · 1.2 KB
/
PanopticFCN-R50+FaPN-1x-FAST.yaml
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
MODEL:
META_ARCHITECTURE: "PanopticFCN"
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
# WEIGHTS: "/raid/huangsh/codes/detectron2/projects/PanopticFCN/output/model_final.pth"
# WEIGHTS: "/data/detectron2_modelzoo/basemodel/R-50-MSRA.pkl"
MASK_ON: True
PIXEL_MEAN: [123.675, 116.28, 103.53]
PIXEL_STD: [1.0, 1.0, 1.0]
RESNETS:
DEPTH: 50
OUT_FEATURES: ["res2", "res3", "res4", "res5"]
FPN:
IN_FEATURES: ["res2", "res3", "res4", "res5"]
POSITION_HEAD:
THING:
THRES: 0.2
STUFF:
THRES: 0.1
DATASETS:
TRAIN: ("coco_2017_train_panoptic_separated",)
TEST: ("coco_2017_val_panoptic_separated",)
DATALOADER:
FILTER_EMPTY_ANNOTATIONS: True
SOLVER:
BASE_LR: 0.01
WEIGHT_DECAY: 1e-4
LR_SCHEDULER_NAME: "WarmupPolyLR"
POLY_LR_POWER: 0.9
WARMUP_ITERS: 1000
WARMUP_FACTOR: 0.001
WARMUP_METHOD: "linear"
CLIP_GRADIENTS:
ENABLED: True
CLIP_VALUE: 35.0
IMS_PER_BATCH: 16
MAX_ITER: 90000
CHECKPOINT_PERIOD: 10000
INPUT:
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
MIN_SIZE_TRAIN_SAMPLING: "choice"
MIN_SIZE_TEST: 800
MAX_SIZE_TRAIN: 1333
MAX_SIZE_TEST: 1333
MASK_FORMAT: "bitmask"
VERSION: 2
OUTPUT_DIR: "./PanopticFCN_r50_1x-Fast_FAM"