Update __init__

Gradiant · Aug 20, 2024 · 877eb38 · 877eb38
1 parent 737880c
commit 877eb38
Showing 1 changed file with 9 additions and 1 deletion.
diff --git 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'
 ]