Update input parameter

Author: SWHL

label_convert/coco_to_labelImg.py

@@ -15,6 +15,9 @@
 
 class COCOTolabelImg:
     def __init__(self, data_dir: ValueType = None, save_dir: ValueType = None):
+        if data_dir is None:
+            raise ValueError("data_dir must not be None")
+
         self.data_dir = Path(data_dir)
         self.verify_exists(self.data_dir)
 

label_convert/darknet_to_coco.py

@@ -20,6 +20,9 @@ def __init__(
         data_dir: ValueType = None,
         save_dir: ValueType = None,
     ):
+        if data_dir is None:
+            raise ValueError("data_dir must not be None")
+
         self.data_dir = Path(data_dir)
         self.verify_exists(self.data_dir)
 

label_convert/labelImg_to_publaynet.py

@@ -23,6 +23,9 @@ def __init__(
         have_test: bool = True,
         test_ratio: float = 0.2,
     ):
+        if data_dir is None:
+            raise ValueError("data_dir must not be None")
+
         self.data_dir = Path(data_dir)
         self.verify_exists(data_dir)
 
@@ -132,7 +135,7 @@ def generate_json(
             label_path = img_path.with_name(f"{img_path.stem}.txt")
             try:
                 label_datas = self.read_txt(label_path)
-            except Exception as e:
+            except Exception:
                 print(f"{label_path} meets error.")
                 continue
 
@@ -204,16 +207,17 @@ def mkdir(dir_path: Union[Path, str]) -> None:
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument(
-        "--data_dir", type=str, required=True, help="The directory from labelImg."
-    )
+    parser.add_argument("--data_dir", type=str, default=None)
+    parser.add_argument("--save_dir", type=str, default=None)
     parser.add_argument("--val_ratio", type=float, default=0.2)
     parser.add_argument("--have_test", action="store_true", default=False)
     parser.add_argument("--test_ratio", type=float, default=0.2)
     args = parser.parse_args()
 
-    converter = LabelImgToPubLayNet(args.val_ratio, args.have_test, args.test_ratio)
-    converter(args.data_dir)
+    converter = LabelImgToPubLayNet(
+        args.data_dir, args.save_dir, args.val_ratio, args.have_test, args.test_ratio
+    )
+    converter()
     print(f"Successfully output to the {args.out_dir}")
 
 

label_convert/labelImg_to_yolov5.py

@@ -22,6 +22,9 @@ def __init__(
         have_test: bool = False,
         test_ratio: float = 0.2,
     ):
+        if data_dir is None:
+            raise ValueError("data_dir must not be None")
+
         self.data_dir = Path(data_dir)
         self.verify_exists(self.data_dir)
 

label_convert/labelme_to_coco.py

@@ -7,7 +7,7 @@
 import shutil
 import time
 from pathlib import Path
-from typing import List, Optional, Union
+from typing import Union
 
 import numpy as np
 from tqdm import tqdm
@@ -16,7 +16,6 @@
 
 
 class LabelmeToCOCO:
-
     def __init__(
         self,
         data_dir: ValueType = None,
@@ -25,7 +24,10 @@ def __init__(
         have_test: bool = False,
         test_ratio: float = 0.2,
     ):
+        if data_dir is None:
+            raise ValueError("data_dir must not be None")
         self.data_dir = Path(data_dir)
+        self.verify_exists(self.data_dir)
 
         self.val_ratio = val_ratio
         self.test_ratio = test_ratio

label_convert/yolov5_to_coco.py

@@ -18,8 +18,10 @@
 
 class YOLOV5ToCOCO:
     def __init__(self, data_dir: ValueType = None, save_dir: ValueType = None):
+        if data_dir is None:
+            raise ValueError("data_dir must not be None")
         self.data_dir = Path(data_dir)
-
+        self.verify_exists(self.data_dir)
         self.verify_exists(self.data_dir / "images")
         self.verify_exists(self.data_dir / "labels")