models/datasets/pipelines/__init__.py

from .top_down_transform import (TopDownAffineFewShot,
                                 TopDownGenerateTargetFewShot)

__all__ = [
    'TopDownGenerateTargetFewShot', 'TopDownAffineFewShot'
]