[Bug] Revise flip transformation in some conditional gan's setting (#…

…1331) revise flip transformation in some conditional gan's setting
open-mmlab · Oct 19, 2022 · d1725ae · d1725ae
1 parent 1e741ea
commit d1725ae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/configs/_base_/datasets/cifar10_nopad.py b/configs/_base_/datasets/cifar10_nopad.py
@@ -1,7 +1,5 @@
-# custom_imports = dict(
-#     imports=['mmcls.datasets.transforms'], allow_failed_imports=False)
 cifar_pipeline = [
-    dict(type='Flip', flip_ratio=0.5, direction='horizontal'),
+    dict(type='Flip', keys=['img'], flip_ratio=0.5, direction='horizontal'),
     dict(type='PackEditInputs')
 ]
 cifar_dataset = dict(

diff --git a/configs/_base_/datasets/imagenet_128.py b/configs/_base_/datasets/imagenet_128.py
@@ -7,7 +7,7 @@
     dict(type='LoadImageFromFile', key='img'),
     dict(type='RandomCropLongEdge', keys=['img']),
     dict(type='Resize', scale=(128, 128), keys=['img'], backend='pillow'),
-    dict(type='Flip', flip_ratio=0.5, direction='horizontal'),
+    dict(type='Flip', keys=['img'], flip_ratio=0.5, direction='horizontal'),
     dict(type='PackEditInputs')
 ]