Merge pull request #4 from dribnet/valid_dir

Add valid dir when necessary.
main
Alex J. Champandard 9 years ago committed by GitHub
commit fcc5e87858

@ -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])

Loading…
Cancel
Save