mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Increase scale of the patch selector random distribution

Browse Source 
This will cause larger slices of an image to appear more frequently,
increasing the difficulty of the generator.
This commit is contained in:
James Betker 2020-09-16 08:27:42 -06:00
parent f211575e9d
commit 0b047e5f80
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/data/full_image_dataset.py
View File

@ -97,7 +97,7 @@ class FullImageDataset(data.Dataset):
target_sz = self.opt['min_tile_size'] target_sz = self.opt['min_tile_size']
h, w, _ = image.shape h, w, _ = image.shape
possible_sizes_above_target = h - target_sz possible_sizes_above_target = h - target_sz
square_size = int(target_sz + possible_sizes_above_target * min(np.abs(np.random.normal(scale=.1)), 1.0)) square_size = int(target_sz + possible_sizes_above_target * min(np.abs(np.random.normal(scale=.2)), 1.0))
# Pick the left,top coords to draw the patch from # Pick the left,top coords to draw the patch from
left = self.pick_along_range(w, square_size, .3) left = self.pick_along_range(w, square_size, .3)
top = self.pick_along_range(w, square_size, .3) top = self.pick_along_range(w, square_size, .3)