From f6c325c2d3c9c4dab30f86d7fee33f9af3fe4f86 Mon Sep 17 00:00:00 2001 From: SWHL Date: Wed, 24 Apr 2024 15:26:53 +0800 Subject: [PATCH] Unify description --- label_convert/darknet_to_coco.py | 2 +- label_convert/labelImg_to_publaynet.py | 2 +- label_convert/labelImg_to_yolov5.py | 6 +++--- label_convert/labelme_to_coco.py | 2 +- label_convert/yolov5_to_coco.py | 6 +++--- label_convert/yolov5_to_yolov8.py | 6 +++--- label_convert/yolov5_yaml_to_coco.py | 6 +++--- label_convert/yolov8_to_yolov5.py | 6 +++--- tests/test_labelImg_to_yolov5.py | 4 ++-- tests/test_yolov5_to_coco.py | 4 ++-- tests/test_yolov5_to_yolov8.py | 4 ++-- tests/test_yolov5_yaml_to_coco.py | 4 ++-- tests/test_yolov8_to_yolov5.py | 4 ++-- 13 files changed, 28 insertions(+), 28 deletions(-) diff --git a/label_convert/darknet_to_coco.py b/label_convert/darknet_to_coco.py index 0514d84..6aef637 100644 --- a/label_convert/darknet_to_coco.py +++ b/label_convert/darknet_to_coco.py @@ -217,7 +217,7 @@ def read_txt(txt_path: str) -> List: def main(): - parser = argparse.ArgumentParser() + parser = argparse.ArgumentParser("Datasets convert from darknet to COCO") parser.add_argument( "--data_dir", default="dataset/darknet_dataset", diff --git a/label_convert/labelImg_to_publaynet.py b/label_convert/labelImg_to_publaynet.py index b8e5f1a..a32110b 100644 --- a/label_convert/labelImg_to_publaynet.py +++ b/label_convert/labelImg_to_publaynet.py @@ -206,7 +206,7 @@ def mkdir(dir_path: Union[Path, str]) -> None: def main(): - parser = argparse.ArgumentParser() + parser = argparse.ArgumentParser("Datasets convert from labelImg to PubLayNet") 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) diff --git a/label_convert/labelImg_to_yolov5.py b/label_convert/labelImg_to_yolov5.py index 5ef2c51..e9d68ab 100644 --- a/label_convert/labelImg_to_yolov5.py +++ b/label_convert/labelImg_to_yolov5.py @@ -13,7 +13,7 @@ ValueType = Union[str, Path, None] -class LabelImgToYOLOV5: +class LabelImgToYOLOv5: def __init__( self, data_dir: ValueType = None, @@ -142,7 +142,7 @@ def cp_file(self, file_path: Path, dst_dir: Path): def main(): - parser = argparse.ArgumentParser() + parser = argparse.ArgumentParser("Datasets convert from labelImg to YOLOv5") 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) @@ -150,7 +150,7 @@ def main(): parser.add_argument("--test_ratio", type=float, default=0.2) args = parser.parse_args() - converter = LabelImgToYOLOV5( + converter = LabelImgToYOLOv5( args.data_dir, args.save_dir, args.val_ratio, args.have_test, args.test_ratio ) converter() diff --git a/label_convert/labelme_to_coco.py b/label_convert/labelme_to_coco.py index a29a0d0..25bb60e 100644 --- a/label_convert/labelme_to_coco.py +++ b/label_convert/labelme_to_coco.py @@ -243,7 +243,7 @@ def cp_file(self, file_path: Path, dst_dir: Path): def main(): - parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO") + parser = argparse.ArgumentParser("Datasets converter from labelme to COCO") 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) diff --git a/label_convert/yolov5_to_coco.py b/label_convert/yolov5_to_coco.py index 3d7cdcf..013ddc3 100644 --- a/label_convert/yolov5_to_coco.py +++ b/label_convert/yolov5_to_coco.py @@ -16,7 +16,7 @@ ValueType = Union[str, Path, None] -class YOLOV5ToCOCO: +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") @@ -248,7 +248,7 @@ def write_json(json_path, content: dict): def main(): - parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO") + parser = argparse.ArgumentParser("Datasets converter from YOLOv5 to COCO") parser.add_argument( "--data_dir", type=str, @@ -261,7 +261,7 @@ def main(): ) args = parser.parse_args() - converter = YOLOV5ToCOCO(args.data_dir, args.save_dir) + converter = YOLOv5ToCOCO(args.data_dir, args.save_dir) converter(mode_list=args.mode_list.split(",")) diff --git a/label_convert/yolov5_to_yolov8.py b/label_convert/yolov5_to_yolov8.py index ebe0b1f..0cf8b3a 100644 --- a/label_convert/yolov5_to_yolov8.py +++ b/label_convert/yolov5_to_yolov8.py @@ -11,7 +11,7 @@ ValueType = Union[str, Path, None] -class YOLOV5ToYOLOV8: +class YOLOv5ToYOLOv8: def __init__(self, data_dir: ValueType = None, save_dir: ValueType = None): if data_dir is None: raise ValueError("data_dir must not be None") @@ -77,7 +77,7 @@ def verify_exists(file_path: Union[Path, str]): def main(): - parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO") + parser = argparse.ArgumentParser("Datasets converter from YOLOv5 to YOLOv8") parser.add_argument( "--data_dir", type=str, @@ -90,7 +90,7 @@ def main(): ) args = parser.parse_args() - converter = YOLOV5ToYOLOV8(args.data_dir, args.save_dir) + converter = YOLOv5ToYOLOv8(args.data_dir, args.save_dir) converter(mode_list=args.mode_list.split(",")) diff --git a/label_convert/yolov5_yaml_to_coco.py b/label_convert/yolov5_yaml_to_coco.py index e3b16f0..4e9b018 100644 --- a/label_convert/yolov5_yaml_to_coco.py +++ b/label_convert/yolov5_yaml_to_coco.py @@ -15,7 +15,7 @@ from tqdm import tqdm -class YOLOV5CfgToCOCO: +class YOLOv5CfgToCOCO: def __init__( self, yaml_path: Union[str, Path, None] = None, @@ -256,7 +256,7 @@ def verify_exists(file_path): def main(): - parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO") + parser = argparse.ArgumentParser("Datasets converter from YOLOv5 YAML to COCO") parser.add_argument( "--yaml_path", type=str, @@ -266,7 +266,7 @@ def main(): parser.add_argument("--save_dir", type=str, default=None) args = parser.parse_args() - converter = YOLOV5CfgToCOCO(args.yaml_path, args.save_dir) + converter = YOLOv5CfgToCOCO(args.yaml_path, args.save_dir) converter() diff --git a/label_convert/yolov8_to_yolov5.py b/label_convert/yolov8_to_yolov5.py index f8e7a70..82011bc 100644 --- a/label_convert/yolov8_to_yolov5.py +++ b/label_convert/yolov8_to_yolov5.py @@ -12,7 +12,7 @@ ValueType = Union[str, Path, None] -class YOLOV8ToYOLOV5: +class YOLOv8ToYOLOv5: def __init__( self, data_dir: ValueType = None, @@ -108,7 +108,7 @@ def read_yaml(yaml_path): def main(): - parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO") + parser = argparse.ArgumentParser("Datasets converter from YOLOv8 to YOLOv5") parser.add_argument( "--data_dir", type=str, @@ -122,7 +122,7 @@ def main(): parser.add_argument("--yaml_path", type=str, default=None) args = parser.parse_args() - converter = YOLOV8ToYOLOV5(args.data_dir, args.save_dir, args.yaml_path) + converter = YOLOv8ToYOLOv5(args.data_dir, args.save_dir, args.yaml_path) converter(mode_list=args.mode_list.split(",")) diff --git a/tests/test_labelImg_to_yolov5.py b/tests/test_labelImg_to_yolov5.py index 28a4c9f..45d2ced 100644 --- a/tests/test_labelImg_to_yolov5.py +++ b/tests/test_labelImg_to_yolov5.py @@ -10,7 +10,7 @@ sys.path.append(str(root_dir)) -from label_convert.labelImg_to_yolov5 import LabelImgToYOLOV5 +from label_convert.labelImg_to_yolov5 import LabelImgToYOLOv5 test_file_dir = cur_dir / "test_files" @@ -20,7 +20,7 @@ def test_normal(): data_dir = test_file_dir / dataset_name - convert = LabelImgToYOLOV5(data_dir) + convert = LabelImgToYOLOv5(data_dir) convert() save_dir = test_file_dir / f"{dataset_name}_yolov5" diff --git a/tests/test_yolov5_to_coco.py b/tests/test_yolov5_to_coco.py index 42c92a4..732ac72 100644 --- a/tests/test_yolov5_to_coco.py +++ b/tests/test_yolov5_to_coco.py @@ -10,7 +10,7 @@ sys.path.append(str(root_dir)) -from label_convert.yolov5_to_coco import YOLOV5ToCOCO +from label_convert.yolov5_to_coco import YOLOv5ToCOCO test_file_dir = cur_dir / "test_files" @@ -19,7 +19,7 @@ def test_normal(): data_dir = test_file_dir / data_dir_name - converter = YOLOV5ToCOCO(data_dir) + converter = YOLOv5ToCOCO(data_dir) converter() save_dir: Path = test_file_dir / f"{data_dir_name}_coco" diff --git a/tests/test_yolov5_to_yolov8.py b/tests/test_yolov5_to_yolov8.py index 26b7b47..1e345c7 100644 --- a/tests/test_yolov5_to_yolov8.py +++ b/tests/test_yolov5_to_yolov8.py @@ -10,7 +10,7 @@ sys.path.append(str(root_dir)) -from label_convert.yolov5_to_yolov8 import YOLOV5ToYOLOV8 +from label_convert.yolov5_to_yolov8 import YOLOv5ToYOLOv8 test_file_dir = cur_dir / "test_files" @@ -19,7 +19,7 @@ def test_normal(): data_dir = test_file_dir / data_dir_name - converter = YOLOV5ToYOLOV8(data_dir) + converter = YOLOv5ToYOLOv8(data_dir) converter() save_dir: Path = test_file_dir / f"{data_dir_name}_yolov8" diff --git a/tests/test_yolov5_yaml_to_coco.py b/tests/test_yolov5_yaml_to_coco.py index cbea6d0..c797df8 100644 --- a/tests/test_yolov5_yaml_to_coco.py +++ b/tests/test_yolov5_yaml_to_coco.py @@ -10,7 +10,7 @@ sys.path.append(str(root_dir)) -from label_convert.yolov5_yaml_to_coco import YOLOV5CfgToCOCO +from label_convert.yolov5_yaml_to_coco import YOLOv5CfgToCOCO test_file_dir = cur_dir / "test_files" @@ -19,7 +19,7 @@ def test_normal(): yaml_path = test_file_dir / data_dir_name / "sample.yaml" - converter = YOLOV5CfgToCOCO(yaml_path=yaml_path) + converter = YOLOv5CfgToCOCO(yaml_path=yaml_path) converter() save_dir: Path = test_file_dir / f"{data_dir_name}_coco" diff --git a/tests/test_yolov8_to_yolov5.py b/tests/test_yolov8_to_yolov5.py index 39d7352..debd18c 100644 --- a/tests/test_yolov8_to_yolov5.py +++ b/tests/test_yolov8_to_yolov5.py @@ -10,7 +10,7 @@ sys.path.append(str(root_dir)) -from label_convert.yolov8_to_yolov5 import YOLOV8ToYOLOV5 +from label_convert.yolov8_to_yolov5 import YOLOv8ToYOLOv5 test_file_dir = cur_dir / "test_files" @@ -19,7 +19,7 @@ def test_normal(): data_dir = test_file_dir / data_dir_name - converter = YOLOV8ToYOLOV5(data_dir) + converter = YOLOv8ToYOLOv5(data_dir) converter() save_dir: Path = test_file_dir / f"{data_dir_name}_yolov5"