add flip in cyclegan (#331)

019ff1a5 · Yuxin Wu · 1df9e4a3 · 019ff1a5
Commit 019ff1a5 authored Jul 17, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

examples/GAN/CycleGAN.py examples/GAN/CycleGAN.py +1 -0

No files found.
--- a/examples/GAN/CycleGAN.py
+++ b/examples/GAN/CycleGAN.py
@@ -161,6 +161,7 @@ def get_data(datadir, isTrain=True):
        augs = [
            imgaug.Resize(int(SHAPE * 1.12)),
            imgaug.RandomCrop(SHAPE),
+            imgaug.Flip(horiz=True),
        ]
    else:
        augs = [imgaug.Resize(SHAPE)]