From 559e66bbb26c8520ef3e69e86cd101f4289b9861 Mon Sep 17 00:00:00 2001 From: "Alex J. Champandard" Date: Sat, 12 Nov 2016 15:47:00 +0100 Subject: [PATCH] Fix exception when training with JPEG compression. --- enhance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhance.py b/enhance.py index 585d85e..5d704d3 100755 --- a/enhance.py +++ b/enhance.py @@ -182,7 +182,7 @@ class DataLoader(threading.Thread): seed = seed.resize((orig.size[0]//args.zoom, orig.size[1]//args.zoom), resample=PIL.Image.LANCZOS) if len(args.train_jpeg) > 0: buffer, rng = io.BytesIO(), args.train_jpeg[-1] if len(args.train_jpeg) > 1 else 15 - seed.save(buffer, format='jpeg', quality=args.train_jpeg+random.randrange(-rng, +rng)) + seed.save(buffer, format='jpeg', quality=args.train_jpeg[0]+random.randrange(-rng, +rng)) seed = PIL.Image.open(buffer) orig = scipy.misc.fromimage(orig).astype(np.float32)