mmclassification/mmcls/structures/__init__.py

10 lines
328 B
Python
Raw Normal View History

2022-05-07 18:01:08 +08:00
# 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)
2022-05-07 18:01:08 +08:00
__all__ = [
'ClsDataSample', 'batch_label_to_onehot', 'cat_batch_labels',
'stack_batch_scores', 'tensor_split'
]