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

Merge pull request #4 from dribnet/valid_dir

Browse Source 
Add valid dir when necessary.
This commit is contained in:
Alex J. Champandard
2016-10-30 17:06:33 +01:00
committed by GitHub
parent 1478977f18 c5053806bd
commit fcc5e87858
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
@@ -433,6 +433,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])