forked from facebookresearch/DensePose
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathrpn_densepose_only_R-50-FPN_1x.yaml
33 lines (33 loc) · 1.04 KB
/
rpn_densepose_only_R-50-FPN_1x.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
MODEL:
TYPE: generalized_rcnn
CONV_BODY: FPN.add_fpn_ResNet50_conv5_body
NUM_CLASSES: 2
RPN_ONLY: True
NUM_GPUS: 8
SOLVER:
WEIGHT_DECAY: 0.0001
LR_POLICY: steps_with_decay
BASE_LR: 0.02
GAMMA: 0.1
MAX_ITER: 90000
STEPS: [0, 60000, 80000]
FPN:
FPN_ON: True
MULTILEVEL_RPN: True
RPN_MAX_LEVEL: 6
RPN_MIN_LEVEL: 2
RPN_ANCHOR_START_SIZE: 32
RPN_ASPECT_RATIOS: (0.5, 1, 2)
TRAIN:
WEIGHTS: https://dl.fbaipublicfiles.com/detectron/ImageNetPretrained/MSRA/R-50.pkl
DATASETS: ('keypoints_coco_2014_train', 'keypoints_coco_2014_valminusminival')
SCALES: (800,)
MAX_SIZE: 1333
TEST:
WEIGHTS: https://dl.fbaipublicfiles.com/detectron/35998996/12_2017_baselines/rpn_person_only_R-50-FPN_1x.yaml.08_10_08.0ZWmJm6F/output/train/keypoints_coco_2014_train%3Akeypoints_coco_2014_valminusminival/generalized_rcnn/model_final.pkl
DATASETS: ('dense_coco_2014_minival', 'dense_coco_2014_train', 'dense_coco_2014_valminusminival')
SCALE: 800
MAX_SIZE: 1333
RPN_PRE_NMS_TOP_N: 1000 # Per FPN level
RPN_POST_NMS_TOP_N: 2000
OUTPUT_DIR: .