Merge pull request #4 from dribnet/valid_dir

Add valid dir when necessary.
2016-10-30 17:06:33 +01:00
parent 1478977f18 c5053806bd
commit fcc5e87858
1 changed files with 1 additions and 0 deletions
@@ -433,6 +433,7 @@ class NeuralEnhancer(object):
        image.save(fn)

    def show_progress(self, orign, scald, repro):
+        os.makedirs('valid', exist_ok=True)
        for i in range(args.batch_size):
            self.imsave('valid/%03i_origin.png' % i, orign[i])
            self.imsave('valid/%03i_pixels.png' % i, scald[i])