Ailusion-VTON-DEMO-v1.1 / configs /cse /densepose_rcnn_R_101_FPN_soft_s1x.yaml
Roopansh's picture
Initial Commit
f981a9d
raw history blame
No virus
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)