From 8bc3cbb8b12908a761b405457868cad48046b9d7 Mon Sep 17 00:00:00 2001
From: Christopher Hesse <christopher.hesse@gmail.com>
Date: Thu, 26 Jan 2017 22:14:20 -0800
Subject: [PATCH] check input dir for valid files
---
pix2pix.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/pix2pix.py b/pix2pix.py
index 8cc727b..a556bbd 100644
--- a/pix2pix.py
+++ b/pix2pix.py
@@ -199,12 +199,18 @@ def lab_to_rgb(lab):
def load_examples():
+ if not os.path.exists(a.input_dir):
+ raise Exception("input_dir does not exist")
+
input_paths = glob.glob(os.path.join(a.input_dir, "*.jpg"))
decode = tf.image.decode_jpeg
if len(input_paths) == 0:
input_paths = glob.glob(os.path.join(a.input_dir, "*.png"))
decode = tf.image.decode_png
+ if len(input_paths) == 0:
+ raise Exception("input_dir contains no image files")
+
def get_name(path):
name, _ = os.path.splitext(os.path.basename(path))
return name
--
GitLab