Virtual-Try-On / configs /cse /densepose_rcnn_R_101_FPN_soft_s1x.yaml
IDM-VTON
update IDM-VTON Demo
938e515
raw
history blame
298 Bytes
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
RESNETS:
DEPTH: 101
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseV1ConvXHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)