mmpretrain/mmcls/structures/__init__.py

10 lines
328 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
from .cls_data_sample import ClsDataSample
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'
]