Merge pull request #10 from Hitbee-dev/bugfix

🐛 Bugfix: save_dir TypeError
RapidAI · Mar 28, 2024 · e5f34f0 · e5f34f0
2 parents 9c1a391 + 1c86d64
commit e5f34f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/label_convert/coco_to_labelImg.py b/label_convert/coco_to_labelImg.py
@@ -29,7 +29,7 @@ def __init__(self, data_dir: Optional[str] = None, save_dir: Optional[str] = Non
         self.verify_exists(self.train2017_dir)
         self.verify_exists(self.val2017_dir)
 
-        self.save_dir = save_dir
+        self.save_dir = Path(save_dir)
         if save_dir is None:
             self.save_dir = (
                 self.data_dir.parent / f"{self.data_dir.name}_labelImg_format"