mmcls/structures/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .cls_data_sample import ClsDataSample
from .multi_task_data_sample import MultiTaskDataSample
from .utils import (batch_label_to_onehot, cat_batch_labels,
                    stack_batch_scores, tensor_split)

__all__ = [
    'ClsDataSample', 'batch_label_to_onehot', 'cat_batch_labels',
    'stack_batch_scores', 'tensor_split', 'MultiTaskDataSample'
]