diff --git a/enhance.py b/enhance.py old mode 100644 new mode 100755 index 20e4d2c..bc40cdb --- a/enhance.py +++ b/enhance.py @@ -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])