Permalink
Browse files

small bug

  • Loading branch information...
Hantian Zhang
Hantian Zhang committed Aug 18, 2017
1 parent 21323cf commit 028737c8dc1300f1bf368f17c71f061e44199bd8
Showing with 2 additions and 2 deletions.
  1. +1 −1 config.py
  2. +1 −1 train.py
View
@@ -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
View
@@ -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:

0 comments on commit 028737c

Please sign in to comment.