Ailusion-VTON-DEMO-v1.1 / configs /cse /densepose_rcnn_R_50_FPN_DL_soft_s1x.yaml
Roopansh's picture
Initial Commit
f981a9d
raw history blame
No virus
296 Bytes
_BASE_: "Base-DensePose-RCNN-FPN-Human.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
RESNETS:
DEPTH: 50
ROI_DENSEPOSE_HEAD:
NAME: "DensePoseDeepLabHead"
CSE:
EMBED_LOSS_NAME: "SoftEmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)