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

Add valid dir when necessary

Browse Source
This commit is contained in:
Tom White
2016-10-31 00:05:37 +13:00
parent f68f04fb1c
commit c5053806bd
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
enhance.py
Regular → Executable
+1
View File
@@ -434,6 +434,7 @@ class NeuralEnhancer(object):
image.save(fn) image.save(fn)
def show_progress(self, orign, scald, repro): def show_progress(self, orign, scald, repro):
os.makedirs('valid', exist_ok=True)
for i in range(args.batch_size): for i in range(args.batch_size):
self.imsave('valid/%03i_origin.png' % i, orign[i]) self.imsave('valid/%03i_origin.png' % i, orign[i])
self.imsave('valid/%03i_pixels.png' % i, scald[i]) self.imsave('valid/%03i_pixels.png' % i, scald[i])