# Copyright (c) OpenMMLab. All rights reserved. | |
from .topdown_heatmap_base_head import TopdownHeatmapBaseHead | |
from .topdown_heatmap_multi_stage_head import (TopdownHeatmapMSMUHead, | |
TopdownHeatmapMultiStageHead) | |
from .topdown_heatmap_simple_head import TopdownHeatmapSimpleHead | |
from .poseur_head import Poseur_noise_sample | |
__all__ = [ | |
'TopdownHeatmapSimpleHead', 'TopdownHeatmapMultiStageHead', | |
'TopdownHeatmapMSMUHead', 'TopdownHeatmapBaseHead', | |
] | |