|
|
|
|
|
|
|
|
|
from typing import Any
|
|
|
|
from .base import BaseConverter
|
|
|
|
|
|
class HFlipConverter(BaseConverter):
|
|
"""
|
|
Converts various DensePose predictor outputs to DensePose results.
|
|
Each DensePose predictor output type has to register its convertion strategy.
|
|
"""
|
|
|
|
registry = {}
|
|
dst_type = None
|
|
|
|
@classmethod
|
|
|
|
|
|
def convert(cls, predictor_outputs: Any, transform_data: Any, *args, **kwargs):
|
|
"""
|
|
Performs an horizontal flip on DensePose predictor outputs.
|
|
Does recursive lookup for base classes, so there's no need
|
|
for explicit registration for derived classes.
|
|
|
|
Args:
|
|
predictor_outputs: DensePose predictor output to be converted to BitMasks
|
|
transform_data: Anything useful for the flip
|
|
Return:
|
|
An instance of the same type as predictor_outputs
|
|
"""
|
|
return super(HFlipConverter, cls).convert(
|
|
predictor_outputs, transform_data, *args, **kwargs
|
|
)
|
|
|
