From e2fd1d3b794a41c6b509ab51c2d857a23c536741 Mon Sep 17 00:00:00 2001
From: kaminyou ho <kaminyouho@aetherai.com>
Date: Wed, 19 Jul 2023 09:12:57 +0000
Subject: [PATCH] fix: auto create save dir

---
 tools/dataset/semi_coco.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tools/dataset/semi_coco.py b/tools/dataset/semi_coco.py
index f5a01d2..fd61698 100644
--- a/tools/dataset/semi_coco.py
+++ b/tools/dataset/semi_coco.py
@@ -46,9 +46,10 @@ def _save_anno(name, images, annotations):
         new_anno["licenses"] = anno["licenses"]
         new_anno["categories"] = anno["categories"]
         new_anno["info"] = anno["info"]
-        path = "{}/{}".format(SAVE_DIR, "semi_supervised")
+
+        path = os.path.join(SAVE_DIR, "semi_supervised")
         if not os.path.exists(path):
-            os.mkdir(path)
+            os.makedirs(path, exist_ok=True)
 
         with open(
             "{root}/{folder}/{save_name}.json".format(