create folder when necessary

Author: foamliu

.gitignore

@@ -1,3 +1,7 @@
 .idea/
 data/
 models/
+__pycache__/
+custom_layers/__pycache__/
+label_dict.txt
+logs/

demo.py

@@ -33,6 +33,9 @@
     num_samples = 20
     samples = random.sample(test_images, num_samples)
 
+    if not os.path.exists('images'):
+        os.makedirs('images')
+
     for i in range(len(samples)):
         image_name = samples[i]
         filename = os.path.join(test_path, image_name)