IDM-VTON / densepose /modeling /predictors /cse_with_confidence.py
IDM-VTON
update IDM-VTON Demo
938e515
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