a ne@snddlmZddlmZddlmZGdddeZeGdddZeGdd d ZeGd d d Z d S) ) dataclass)Enum)CfgNodec@seZdZdZdZdZdS)DensePoseUVConfidenceTypea Statistical model type for confidence learning, possible values: - "iid_iso": statistically independent identically distributed residuals with anisotropic covariance - "indep_aniso": statistically independent residuals with anisotropic covariances For details, see: N. Neverova, D. Novotny, A. Vedaldi "Correlated Uncertainty for Learning Dense Correspondences from Noisy Labels", p. 918--926, in Proc. NIPS 2019 iid_isoZ indep_anisoN)__name__ __module__ __qualname____doc__IID_ISOZ INDEP_ANISOr r M/home/alin0222/detectron2/projects/DensePose/densepose/modeling/confidence.pyr s rc@s8eZdZUdZdZeed<dZeed<e j Z e ed<dS)DensePoseUVConfidenceConfigz9 Configuration options for confidence on UV data Fenabled{Gz?epsilontypeN) rrr r rbool__annotations__rfloatrr rr r r r rs   rc@s*eZdZUdZdZeed<dZeed<dS)DensePoseSegmConfidenceConfigz> Configuration options for confidence on segmentation FrrrN) rrr r rrrrrr r r r r's  rc@s6eZdZUdZeed<eed<eeddddZ dS)DensePoseConfidenceModelConfigz5 Configuration options for confidence models uv_confidencesegm_confidence)cfgreturncCsJtt|jjjj|jjjjt|jjjjdt |jjj j|jjj jddS)N)rrr)rr)rr) rrMODELROI_DENSEPOSE_HEAD UV_CONFIDENCEENABLEDEPSILONrTYPErSEGM_CONFIDENCE)rr r r from_cfg=s    z'DensePoseConfidenceModelConfig.from_cfgN) rrr r rrr staticmethodrr#r r r r r2s rN) dataclassesrenumrdetectron2.configrrrrrr r r r s