Add valid dir when necessary

9 years ago · c5053806bd
parent f68f04fb1c
commit c5053806bd
1 changed files with 1 additions and 0 deletions
--- a/enhance.py
+++ b/enhance.py
@ -434,6 +434,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])