Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit c505380

Browse files
committed
Add valid dir when necessary
1 parent f68f04f commit c505380

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

enhance.py

100644100755
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -434,6 +434,7 @@ def imsave(self, fn, img):
434434
image.save(fn)
435435

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

0 commit comments

Comments
 (0)