fanwuyang/neural-enhance-super-resolution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Buffer chunks of images proportionally to size.

Browse Source
This commit is contained in:
Alex J. Champandard
2016-11-08 17:15:41 +01:00
parent da354f3e55
commit 2a29fc954b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance.py
+1 -1
View File
@@ -188,7 +188,7 @@ class DataLoader(threading.Thread):
if args.train_noise: if args.train_noise:
seed += scipy.random.normal(scale=args.train_noise, size=(seed.shape[0], seed.shape[1], 1)) ** 4.0 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) h = random.randint(0, seed.shape[0] - self.seed_shape)
w = random.randint(0, seed.shape[1] - 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] seed_chunk = seed[h:h+self.seed_shape, w:w+self.seed_shape]