fanwuyang/neural-enhance-super-resolution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix enhancer.process to pass img, seed

Browse Source
This commit is contained in:
Tom White
2016-10-31 02:14:24 +13:00
parent 37cb208374
commit 8f5167d235
1 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance.py
+1 -5
View File
@@ -221,12 +221,8 @@ class Model(object):
def __init__(self): def __init__(self):
self.network = collections.OrderedDict() self.network = collections.OrderedDict()
if args.train:
self.network['img'] = InputLayer((None, 3, None, None)) self.network['img'] = InputLayer((None, 3, None, None))
self.network['seed'] = InputLayer((None, 3, None, None)) self.network['seed'] = InputLayer((None, 3, None, None))
else:
self.network['img'] = InputLayer((None, 3, None, None))
self.network['seed'] = self.network['img']
config, params = self.load_model() config, params = self.load_model()
self.setup_generator(self.last_layer(), config) self.setup_generator(self.last_layer(), config)
@@ -507,7 +503,7 @@ class NeuralEnhancer(object):
def process(self, image): def process(self, image):
img = np.transpose(image / 255.0 - 0.5, (2, 0, 1))[np.newaxis].astype(np.float32) img = np.transpose(image / 255.0 - 0.5, (2, 0, 1))[np.newaxis].astype(np.float32)
*_, repro = self.model.predict(img) *_, repro = self.model.predict(img, img)
repro = np.transpose(repro[0] + 0.5, (1, 2, 0)).clip(0.0, 1.0) repro = np.transpose(repro[0] + 0.5, (1, 2, 0)).clip(0.0, 1.0)
return scipy.misc.toimage(repro * 255.0, cmin=0, cmax=255) return scipy.misc.toimage(repro * 255.0, cmin=0, cmax=255)