HumanSD / mmpose /models /data_preprocessors /data_preprocessor.py
liyy201912's picture
Upload folder using huggingface_hub
cc0dd3c
raw
history blame contribute delete
265 Bytes
# Copyright (c) OpenMMLab. All rights reserved.
from mmengine.model import ImgDataPreprocessor
from mmpose.registry import MODELS
@MODELS.register_module()
class PoseDataPreprocessor(ImgDataPreprocessor):
"""Image pre-processor for pose estimation tasks."""