From 162d592d9e93a5053fe65e7126489dfc6eea7960 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Sat, 22 Aug 2020 17:15:37 -0700 Subject: [PATCH] make random crop in augmentation a bit less severe --- setup.py | 2 +- stylegan2_pytorch/stylegan2_pytorch.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 82ce5e6..629e953 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ name = 'stylegan2_pytorch', packages = find_packages(), scripts=['bin/stylegan2_pytorch'], - version = '0.20.0', + version = '0.20.1', license='GPLv3+', description = 'StyleGan2 in Pytorch', author = 'Phil Wang', diff --git a/stylegan2_pytorch/stylegan2_pytorch.py b/stylegan2_pytorch/stylegan2_pytorch.py index 95315f8..53bc7b1 100644 --- a/stylegan2_pytorch/stylegan2_pytorch.py +++ b/stylegan2_pytorch/stylegan2_pytorch.py @@ -288,7 +288,7 @@ def __init__(self, D, image_size): def forward(self, images, prob = 0., detach = False): if random() < prob: - random_scale = random_float(0.5, 0.9) + random_scale = random_float(0.75, 0.95) images = random_hflip(images, prob=0.5) images = random_crop_and_resize(images, scale = random_scale)