Buffer chunks of images proportionally to size.

9 years ago · 2a29fc954b
parent da354f3e55
commit 2a29fc954b
1 changed files with 1 additions and 1 deletions
--- a/enhance.py
+++ b/enhance.py
@ -188,7 +188,7 @@ class DataLoader(threading.Thread):
        if args.train_noise:
            seed += scipy.random.normal(scale=args.train_noise, size=(seed.shape[0], seed.shape[1], 1)) ** 4.0

-        for _ in range(args.buffer_similar):
+        for _ in range(seed.shape[0] * seed.shape[1] // self.seed_shape * 2):
            h = random.randint(0, seed.shape[0] - self.seed_shape)
            w = random.randint(0, seed.shape[1] - self.seed_shape)
            seed_chunk = seed[h:h+self.seed_shape, w:w+self.seed_shape]