Ma Zerun a05c79e806
[Refactor] Move transforms in mmselfsup to mmpretrain. (#1396)
* [Refactor] Move transforms in mmselfsup to mmpretrain.

* Update transform docs and configs. And register some mmcv transforms in
mmpretrain.

* Fix missing transform wrapper.

* update selfsup transforms

* Fix UT

* Fix UT

* update gaussianblur inconfigs

---------

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-03 15:01:11 +08:00

11 lines
451 B
Python

# Copyright (c) OpenMMLab. All rights reserved.
from .data_sample import DataSample
from .multi_task_data_sample import MultiTaskDataSample
from .utils import (batch_label_to_onehot, cat_batch_labels, format_label,
format_score, label_to_onehot, tensor_split)
__all__ = [
'DataSample', 'batch_label_to_onehot', 'cat_batch_labels', 'tensor_split',
'MultiTaskDataSample', 'label_to_onehot', 'format_label', 'format_score'
]