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