small bug

SpaceML · Aug 18, 2017 · 028737c · 028737c
1 parent 21323cf
commit 028737c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/config.py b/config.py
@@ -5,7 +5,7 @@ class Config:
     output_path = "results"
 
     img_size = 424
-    adjust_size = 500
+    adjust_size = 424
     train_size = 424
     img_channel = 3
     conv_channel_base = 64

diff --git a/train.py b/train.py
@@ -12,7 +12,7 @@ def prepocess_train(img, cond,):
     img = scipy.misc.imresize(img, [conf.adjust_size, conf.adjust_size])
     cond = scipy.misc.imresize(cond, [conf.adjust_size, conf.adjust_size])
     h1 = int(np.ceil(np.random.uniform(1e-2, conf.adjust_size - conf.train_size)))
-    w1 = int(np.ceil(np.random.uniform(1e-2, conf.adjust_size - conf.adjust_size)))
+    w1 = int(np.ceil(np.random.uniform(1e-2, conf.adjust_size - conf.train_size)))
     img = img[h1:h1 + conf.train_size, w1:w1 + conf.train_size]
     cond = cond[h1:h1 + conf.train_size, w1:w1 + conf.train_size]
     if np.random.random() > 0.5: