Spaces:
Running
on
Zero
Running
on
Zero
Virtual-Try-On
/
preprocess
/humanparsing
/mhp_extension
/detectron2
/projects
/DensePose
/densepose
/config.py
# -*- coding = utf-8 -*- | |
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved | |
from detectron2.config import CfgNode as CN | |
def add_dataset_category_config(cfg: CN): | |
""" | |
Add config for additional category-related dataset options | |
- category whitelisting | |
- category mapping | |
""" | |
_C = cfg | |
_C.DATASETS.CATEGORY_MAPS = CN(new_allowed=True) | |
_C.DATASETS.WHITELISTED_CATEGORIES = CN(new_allowed=True) | |
def add_densepose_config(cfg: CN): | |
""" | |
Add config for densepose head. | |
""" | |
_C = cfg | |
_C.MODEL.DENSEPOSE_ON = True | |
_C.MODEL.ROI_DENSEPOSE_HEAD = CN() | |
_C.MODEL.ROI_DENSEPOSE_HEAD.NAME = "" | |
_C.MODEL.ROI_DENSEPOSE_HEAD.NUM_STACKED_CONVS = 8 | |
# Number of parts used for point labels | |
_C.MODEL.ROI_DENSEPOSE_HEAD.NUM_PATCHES = 24 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DECONV_KERNEL = 4 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.CONV_HEAD_DIM = 512 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.CONV_HEAD_KERNEL = 3 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.UP_SCALE = 2 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.HEATMAP_SIZE = 112 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.POOLER_TYPE = "ROIAlignV2" | |
_C.MODEL.ROI_DENSEPOSE_HEAD.POOLER_RESOLUTION = 28 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.POOLER_SAMPLING_RATIO = 2 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.NUM_COARSE_SEGM_CHANNELS = 2 # 15 or 2 | |
# Overlap threshold for an RoI to be considered foreground (if >= FG_IOU_THRESHOLD) | |
_C.MODEL.ROI_DENSEPOSE_HEAD.FG_IOU_THRESHOLD = 0.7 | |
# Loss weights for annotation masks.(14 Parts) | |
_C.MODEL.ROI_DENSEPOSE_HEAD.INDEX_WEIGHTS = 5.0 | |
# Loss weights for surface parts. (24 Parts) | |
_C.MODEL.ROI_DENSEPOSE_HEAD.PART_WEIGHTS = 1.0 | |
# Loss weights for UV regression. | |
_C.MODEL.ROI_DENSEPOSE_HEAD.POINT_REGRESSION_WEIGHTS = 0.01 | |
# For Decoder | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DECODER_ON = True | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DECODER_NUM_CLASSES = 256 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DECODER_CONV_DIMS = 256 | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DECODER_NORM = "" | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DECODER_COMMON_STRIDE = 4 | |
# For DeepLab head | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DEEPLAB = CN() | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DEEPLAB.NORM = "GN" | |
_C.MODEL.ROI_DENSEPOSE_HEAD.DEEPLAB.NONLOCAL_ON = 0 | |
# Confidences | |
# Enable learning confidences (variances) along with the actual values | |
_C.MODEL.ROI_DENSEPOSE_HEAD.UV_CONFIDENCE = CN({"ENABLED": False}) | |
# UV confidence lower bound | |
_C.MODEL.ROI_DENSEPOSE_HEAD.UV_CONFIDENCE.EPSILON = 0.01 | |
# Statistical model type for confidence learning, possible values: | |
# - "iid_iso": statistically independent identically distributed residuals | |
# with isotropic covariance | |
# - "indep_aniso": statistically independent residuals with anisotropic | |
# covariances | |
_C.MODEL.ROI_DENSEPOSE_HEAD.UV_CONFIDENCE.TYPE = "iid_iso" | |