From 0b047e5f80c5ea9c58c4227ae4973d150c6647bf Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 16 Sep 2020 08:27:42 -0600 Subject: [PATCH] Increase scale of the patch selector random distribution This will cause larger slices of an image to appear more frequently, increasing the difficulty of the generator. --- codes/data/full_image_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/data/full_image_dataset.py b/codes/data/full_image_dataset.py index 2ba9b1e2..d36f22d6 100644 --- a/codes/data/full_image_dataset.py +++ b/codes/data/full_image_dataset.py @@ -97,7 +97,7 @@ class FullImageDataset(data.Dataset): target_sz = self.opt['min_tile_size'] h, w, _ = image.shape 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 left = self.pick_along_range(w, square_size, .3) top = self.pick_along_range(w, square_size, .3)