Virtual-Try-Ons / configs /HRNet /densepose_rcnn_HRFPN_HRNet_w40_s1x.yaml
IDM-VTON
update IDM-VTON Demo
938e515
raw
history blame
545 Bytes
_BASE_: "../Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "https://1drv.ms/u/s!Aus8VCZ_C_33ck0gvo5jfoWBOPo"
BACKBONE:
NAME: "build_hrfpn_backbone"
RPN:
IN_FEATURES: ['p1', 'p2', 'p3', 'p4', 'p5']
ROI_HEADS:
IN_FEATURES: ['p1', 'p2', 'p3', 'p4', 'p5']
HRNET:
STAGE2:
NUM_CHANNELS: [40, 80]
STAGE3:
NUM_CHANNELS: [40, 80, 160]
STAGE4:
NUM_CHANNELS: [40, 80, 160, 320]
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)
CLIP_GRADIENTS:
ENABLED: True
CLIP_TYPE: "norm"
BASE_LR: 0.03