Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit fabe04f

Browse files
committed
Fix bug in handling of --train-jpeg parameter.
1 parent 7c8cc56 commit fabe04f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

enhance.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def add_to_buffer(self, f):
180180
seed = seed.filter(PIL.ImageFilter.GaussianBlur(radius=random.randint(0, args.train_blur*2)))
181181
if args.zoom > 1:
182182
seed = seed.resize((orig.size[0]//args.zoom, orig.size[1]//args.zoom), resample=PIL.Image.LANCZOS)
183-
if args.train_jpeg is not None:
183+
if len(args.train_jpeg) > 0:
184184
buffer, rng = io.BytesIO(), args.train_jpeg[-1] if len(args.train_jpeg) > 1 else 15
185185
seed.save(buffer, format='jpeg', quality=args.train_jpeg+random.randrange(-rng, +rng))
186186
seed = PIL.Image.open(buffer)

0 commit comments

Comments
 (0)