Virtual-Try-On / configs /densepose_rcnn_R_50_FPN_WC1M_s1x.yaml
IDM-VTON
update IDM-VTON Demo
938e515
raw
history blame
445 Bytes
_BASE_: "Base-DensePose-RCNN-FPN.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
ROI_DENSEPOSE_HEAD:
UV_CONFIDENCE:
ENABLED: True
TYPE: "iid_iso"
SEGM_CONFIDENCE:
ENABLED: True
POINT_REGRESSION_WEIGHTS: 0.0005
SOLVER:
CLIP_GRADIENTS:
ENABLED: True
CLIP_TYPE: norm
CLIP_VALUE: 100.0
MAX_ITER: 130000
STEPS: (100000, 120000)
WARMUP_FACTOR: 0.025