Option to add distortions to random low-res images

This commit is contained in:
igv
2017-07-27 14:46:28 +03:00
parent f6b024388f
commit 15bc798701
3 changed files with 19 additions and 7 deletions

View File

@@ -33,6 +33,7 @@ class FSRCNN(object):
self.batch_size = config.batch_size
self.learning_rate = config.learning_rate
self.threads = config.threads
self.distort = config.distort
self.params = config.params
# Different image/label sub-sizes for different scaling factors x2, x3, x4