From c5053806bdbfeabedd84d94947f85da10870e2b9 Mon Sep 17 00:00:00 2001 From: Tom White Date: Sat, 29 Oct 2016 23:55:11 +1300 Subject: [PATCH] Add valid dir when necessary --- enhance.py | 1 + 1 file changed, 1 insertion(+) mode change 100644 => 100755 enhance.py diff --git a/enhance.py b/enhance.py old mode 100644 new mode 100755 index e1cfe5e..4c5ed42 --- 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])