From 877eb38c2e56a93a00ab673792a2d902f5af192e Mon Sep 17 00:00:00 2001 From: ccanamero Date: Tue, 20 Aug 2024 16:56:11 +0200 Subject: [PATCH] Update __init__ --- mmcls/datasets/__init__.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) 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' ]