jiovirtualtryon / configs /cse /densepose_rcnn_R_101_FPN_s1x.yaml
atlury's picture
Upload 61 files
268955d verified
raw
history blame
306 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: "EmbeddingLoss"
SOLVER:
MAX_ITER: 130000
STEPS: (100000, 120000)