diff options
Diffstat (limited to 'scripts/builders/mogrify-dir.py')
| -rwxr-xr-x | scripts/builders/mogrify-dir.py | 20 |
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) |