Spaces:
Running
on
Zero
Running
on
Zero
Virtual-Try-On
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/configs
/Detectron1-Comparisons
/keypoint_rcnn_R_50_FPN_1x.yaml
_BASE_: "../Base-RCNN-FPN.yaml" | |
MODEL: | |
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl" | |
KEYPOINT_ON: True | |
RESNETS: | |
DEPTH: 50 | |
ROI_HEADS: | |
NUM_CLASSES: 1 | |
ROI_KEYPOINT_HEAD: | |
POOLER_RESOLUTION: 14 | |
POOLER_SAMPLING_RATIO: 2 | |
POOLER_TYPE: "ROIAlign" | |
# Detectron1 uses smooth L1 loss with some magic beta values. | |
# The defaults are changed to L1 loss in Detectron2. | |
ROI_BOX_HEAD: | |
SMOOTH_L1_BETA: 1.0 | |
POOLER_SAMPLING_RATIO: 2 | |
POOLER_TYPE: "ROIAlign" | |
RPN: | |
SMOOTH_L1_BETA: 0.1111 | |
# Detectron1 uses 2000 proposals per-batch, but this option is per-image in detectron2 | |
# 1000 proposals per-image is found to hurt box AP. | |
# Therefore we increase it to 1500 per-image. | |
POST_NMS_TOPK_TRAIN: 1500 | |
DATASETS: | |
TRAIN: ("keypoints_coco_2017_train",) | |
TEST: ("keypoints_coco_2017_val",) | |