densepose/modeling/predictors/cse_with_confidence.py

# 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