diff --git a/mmcls/datasets/__init__.py b/mmcls/datasets/__init__.py index 83ae84754e..a803763b7d 100644 --- a/mmcls/datasets/__init__.py +++ b/mmcls/datasets/__init__.py @@ -14,6 +14,13 @@ from .samplers import DistributedSampler, RepeatAugSampler from .stanford_cars import StanfordCars from .voc import VOC +from .pipelines.test_time_aug import MultiScaleFlipAug +from .pipelines.manage_multichannel_image import ( + LoadMultiChannelImgFromFile, + ResizeMultiChannel, + BrightnessTransformMultiChannel, + NormalizeMinMaxChannelwise +) __all__ = [ @@ -22,5 +29,6 @@ 'DistributedSampler', 'ConcatDataset', 'RepeatDataset', 'ClassBalancedDataset', 'DATASETS', 'PIPELINES', 'ImageNet21k', 'SAMPLERS', 'build_sampler', 'RepeatAugSampler', 'KFoldDataset', 'CUB', - 'CustomDataset', 'StanfordCars' + 'CustomDataset', 'StanfordCars', 'MultiScaleFlipAug', 'LoadMultiChannelImgFromFile', + 'ResizeMultiChannel', 'BrightnessTransformMultiChannel', 'NormalizeMinMaxChannelwise' ]