Skip to content

S
seminar-breakout
Commit 019ff1a5 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

add flip in cyclegan (#331)

parent 1df9e4a3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/GAN/CycleGAN.py
View file @ 019ff1a5
...@@ -161,6 +161,7 @@ def get_data(datadir, isTrain=True): ...@@ -161,6 +161,7 @@ def get_data(datadir, isTrain=True):
augs = [ augs = [
imgaug.Resize(int(SHAPE * 1.12)), imgaug.Resize(int(SHAPE * 1.12)),
imgaug.RandomCrop(SHAPE), imgaug.RandomCrop(SHAPE),
imgaug.Flip(horiz=True),
] ]
else: else:
augs = [imgaug.Resize(SHAPE)] augs = [imgaug.Resize(SHAPE)]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment