debug image dir

fetch_fivek.py

@@ -57,14 +57,14 @@ def download(url, fn=None, path=None):
   print()
   print()
   print('Congratulations:'
-        '  The MIT-Adobe FiveK Dataset is ready.'
-        '  You can train your own model with \'python3 train.py example test\'.'
+        '  The MIT-Adobe FiveK Dataset is ready.\n'
+        '  You can train your own model with \'python3 train.py example test\'.\n'
         '  Your trained model will be located at \'models/example/test\'')
 
   print()
-  print('Note:'
-        '  Due to copyright issues, we cannot provide photos from 500px artists.'
-        '  If you want to try your own output dataset, please collect your own stylized images,'
+  print('Note:\n'
+        '  Due to copyright issues, we cannot provide photos from 500px artists.\n'
+        '  If you want to try your own output dataset, please collect your own stylized images,\n'
         '  and put them under artists/[ArtistName]/*.{jpg|png}')
 
 

net.py

@@ -32,6 +32,8 @@ def __init__(self, cfg, restore=False):
       os.makedirs(self.dir)
     if not os.path.exists(self.dump_dir):
       os.makedirs(self.dump_dir)
+    if not os.path.exists(self.image_dir):
+      os.makedirs(self.image_dir)
 
     if not restore:
       self.backup_scripts()