Merge branch 'master' of ghghgh.us:pix2pix

1 files changed, 10 insertions, 10 deletions

diff --git a/scripts/builders/mogrify-dir.py b/scripts/builders/mogrify-dir.py
index 2794ee6..4453322 100755
--- a/scripts/builders/mogrify-dir.py
+++ b/scripts/builders/mogrify-dir.py
@@ -26,15 +26,15 @@ if os.path.exists(out_dir):
 
 os.makedirs(out_dir)
 if opt.split is True:
-  os.makedirs(out_dir + "A/")
-  os.makedirs(out_dir + "A/train/")
-  os.makedirs(out_dir + "A/test/")
-  os.makedirs(out_dir + "A/val/")
+  os.makedirs(os.path.join(out_dir, "A/"))
+  os.makedirs(os.path.join(out_dir, "A/train/"))
+  os.makedirs(os.path.join(out_dir, "A/test/"))
+  os.makedirs(os.path.join(out_dir, "A/val/"))
   if opt.ab is True:
-    os.makedirs(out_dir + "B/")
-    os.makedirs(out_dir + "B/train/")
-    os.makedirs(out_dir + "B/test/")
-    os.makedirs(out_dir + "B/val/")
+    os.makedirs(os.path.join(out_dir, "B/"))
+    os.makedirs(os.path.join(out_dir, "B/train/"))
+    os.makedirs(os.path.join(out_dir, "B/test/"))
+    os.makedirs(os.path.join(out_dir, "B/val/"))
 
 file = open(os.path.join(out_dir, "opt.txt"), "w") 
 for arg in vars(opt):
@@ -87,9 +87,9 @@ for i, fn in enumerate(images):
     else:
       wd = "train/"
 
-    cv2.imwrite(out_dir + "A/" + wd + out_file, img)
+    cv2.imwrite(os.path.join(out_dir, "A/", wd, out_file), img)
     if opt.ab is True:
-      copyfile(fn, out_dir + "B/" + wd + out_file)
+      copyfile(fn, os.path.join(out_dir, "B/", wd, out_file))
 
   else:
     cv2.imwrite(os.path.join(out_dir, out_file), img)