Leffa / 3rdparty /densepose /modeling /predictors /cse_with_confidence.py
franciszzj's picture
init code
b213d84
raw
history blame
448 Bytes
# Copyright (c) Facebook, Inc. and its affiliates.
from . import DensePoseEmbeddingConfidencePredictorMixin, DensePoseEmbeddingPredictor
from .registry import DENSEPOSE_PREDICTOR_REGISTRY
@DENSEPOSE_PREDICTOR_REGISTRY.register()
class DensePoseEmbeddingWithConfidencePredictor(
DensePoseEmbeddingConfidencePredictorMixin, DensePoseEmbeddingPredictor
):
"""
Predictor that combines CSE and CSE confidence estimation
"""
pass