Skip to content

Commit f6c325c

Browse files
SWHLSWHL
committed
Unify description
1 parent 72320e4 commit f6c325c

13 files changed

+28
-28
lines changed

‎label_convert/darknet_to_coco.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,7 @@ def read_txt(txt_path: str) -> List:
217217

218218

219219
def main():
220-
parser = argparse.ArgumentParser()
220+
parser = argparse.ArgumentParser("Datasets convert from darknet to COCO")
221221
parser.add_argument(
222222
"--data_dir",
223223
default="dataset/darknet_dataset",

‎label_convert/labelImg_to_publaynet.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,7 @@ def mkdir(dir_path: Union[Path, str]) -> None:
206206

207207

208208
def main():
209-
parser = argparse.ArgumentParser()
209+
parser = argparse.ArgumentParser("Datasets convert from labelImg to PubLayNet")
210210
parser.add_argument("--data_dir", type=str, default=None)
211211
parser.add_argument("--save_dir", type=str, default=None)
212212
parser.add_argument("--val_ratio", type=float, default=0.2)

‎label_convert/labelImg_to_yolov5.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
ValueType = Union[str, Path, None]
1414

1515

16-
class LabelImgToYOLOV5:
16+
class LabelImgToYOLOv5:
1717
def __init__(
1818
self,
1919
data_dir: ValueType = None,
@@ -142,15 +142,15 @@ def cp_file(self, file_path: Path, dst_dir: Path):
142142

143143

144144
def main():
145-
parser = argparse.ArgumentParser()
145+
parser = argparse.ArgumentParser("Datasets convert from labelImg to YOLOv5")
146146
parser.add_argument("--data_dir", type=str, default=None)
147147
parser.add_argument("--save_dir", type=str, default=None)
148148
parser.add_argument("--val_ratio", type=float, default=0.2)
149149
parser.add_argument("--have_test", action="store_true", default=False)
150150
parser.add_argument("--test_ratio", type=float, default=0.2)
151151
args = parser.parse_args()
152152

153-
converter = LabelImgToYOLOV5(
153+
converter = LabelImgToYOLOv5(
154154
args.data_dir, args.save_dir, args.val_ratio, args.have_test, args.test_ratio
155155
)
156156
converter()

‎label_convert/labelme_to_coco.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def cp_file(self, file_path: Path, dst_dir: Path):
243243

244244

245245
def main():
246-
parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO")
246+
parser = argparse.ArgumentParser("Datasets converter from labelme to COCO")
247247
parser.add_argument("--data_dir", type=str, default=None)
248248
parser.add_argument("--save_dir", type=str, default=None)
249249
parser.add_argument("--val_ratio", type=float, default=0.2)

‎label_convert/yolov5_to_coco.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
ValueType = Union[str, Path, None]
1717

1818

19-
class YOLOV5ToCOCO:
19+
class YOLOv5ToCOCO:
2020
def __init__(self, data_dir: ValueType = None, save_dir: ValueType = None):
2121
if data_dir is None:
2222
raise ValueError("data_dir must not be None")
@@ -248,7 +248,7 @@ def write_json(json_path, content: dict):
248248

249249

250250
def main():
251-
parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO")
251+
parser = argparse.ArgumentParser("Datasets converter from YOLOv5 to COCO")
252252
parser.add_argument(
253253
"--data_dir",
254254
type=str,
@@ -261,7 +261,7 @@ def main():
261261
)
262262
args = parser.parse_args()
263263

264-
converter = YOLOV5ToCOCO(args.data_dir, args.save_dir)
264+
converter = YOLOv5ToCOCO(args.data_dir, args.save_dir)
265265
converter(mode_list=args.mode_list.split(","))
266266

267267

‎label_convert/yolov5_to_yolov8.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
ValueType = Union[str, Path, None]
1212

1313

14-
class YOLOV5ToYOLOV8:
14+
class YOLOv5ToYOLOv8:
1515
def __init__(self, data_dir: ValueType = None, save_dir: ValueType = None):
1616
if data_dir is None:
1717
raise ValueError("data_dir must not be None")
@@ -77,7 +77,7 @@ def verify_exists(file_path: Union[Path, str]):
7777

7878

7979
def main():
80-
parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO")
80+
parser = argparse.ArgumentParser("Datasets converter from YOLOv5 to YOLOv8")
8181
parser.add_argument(
8282
"--data_dir",
8383
type=str,
@@ -90,7 +90,7 @@ def main():
9090
)
9191
args = parser.parse_args()
9292

93-
converter = YOLOV5ToYOLOV8(args.data_dir, args.save_dir)
93+
converter = YOLOv5ToYOLOv8(args.data_dir, args.save_dir)
9494
converter(mode_list=args.mode_list.split(","))
9595

9696

‎label_convert/yolov5_yaml_to_coco.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from tqdm import tqdm
1616

1717

18-
class YOLOV5CfgToCOCO:
18+
class YOLOv5CfgToCOCO:
1919
def __init__(
2020
self,
2121
yaml_path: Union[str, Path, None] = None,
@@ -256,7 +256,7 @@ def verify_exists(file_path):
256256

257257

258258
def main():
259-
parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO")
259+
parser = argparse.ArgumentParser("Datasets converter from YOLOv5 YAML to COCO")
260260
parser.add_argument(
261261
"--yaml_path",
262262
type=str,
@@ -266,7 +266,7 @@ def main():
266266
parser.add_argument("--save_dir", type=str, default=None)
267267
args = parser.parse_args()
268268

269-
converter = YOLOV5CfgToCOCO(args.yaml_path, args.save_dir)
269+
converter = YOLOv5CfgToCOCO(args.yaml_path, args.save_dir)
270270
converter()
271271

272272

‎label_convert/yolov8_to_yolov5.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
ValueType = Union[str, Path, None]
1313

1414

15-
class YOLOV8ToYOLOV5:
15+
class YOLOv8ToYOLOv5:
1616
def __init__(
1717
self,
1818
data_dir: ValueType = None,
@@ -108,7 +108,7 @@ def read_yaml(yaml_path):
108108

109109

110110
def main():
111-
parser = argparse.ArgumentParser("Datasets converter from YOLOV5 to COCO")
111+
parser = argparse.ArgumentParser("Datasets converter from YOLOv8 to YOLOv5")
112112
parser.add_argument(
113113
"--data_dir",
114114
type=str,
@@ -122,7 +122,7 @@ def main():
122122
parser.add_argument("--yaml_path", type=str, default=None)
123123
args = parser.parse_args()
124124

125-
converter = YOLOV8ToYOLOV5(args.data_dir, args.save_dir, args.yaml_path)
125+
converter = YOLOv8ToYOLOv5(args.data_dir, args.save_dir, args.yaml_path)
126126
converter(mode_list=args.mode_list.split(","))
127127

128128

‎tests/test_labelImg_to_yolov5.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
sys.path.append(str(root_dir))
1212

13-
from label_convert.labelImg_to_yolov5 import LabelImgToYOLOV5
13+
from label_convert.labelImg_to_yolov5 import LabelImgToYOLOv5
1414

1515
test_file_dir = cur_dir / "test_files"
1616

@@ -20,7 +20,7 @@
2020

2121
def test_normal():
2222
data_dir = test_file_dir / dataset_name
23-
convert = LabelImgToYOLOV5(data_dir)
23+
convert = LabelImgToYOLOv5(data_dir)
2424
convert()
2525

2626
save_dir = test_file_dir / f"{dataset_name}_yolov5"

‎tests/test_yolov5_to_coco.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
sys.path.append(str(root_dir))
1212

13-
from label_convert.yolov5_to_coco import YOLOV5ToCOCO
13+
from label_convert.yolov5_to_coco import YOLOv5ToCOCO
1414

1515
test_file_dir = cur_dir / "test_files"
1616

@@ -19,7 +19,7 @@
1919

2020
def test_normal():
2121
data_dir = test_file_dir / data_dir_name
22-
converter = YOLOV5ToCOCO(data_dir)
22+
converter = YOLOv5ToCOCO(data_dir)
2323
converter()
2424

2525
save_dir: Path = test_file_dir / f"{data_dir_name}_coco"

0 commit comments

Comments
 (0)