Rename to AnnotationsConverter

zhiqwang · May 1, 2022 · 7840363 · 7840363
1 parent 34bb53b
commit 7840363
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tools/convert_yolo_to_coco.py → tools/convert_txt_to_json.py b/tools/convert_yolo_to_coco.py → tools/convert_txt_to_json.py
@@ -2,11 +2,11 @@
 
 import argparse
 
-from yolort.utils.yolo2coco import YOLO2COCO
+from yolort.utils.annotations_converter import AnnotationsConverter
 
 
 def get_parser():
-    parser = argparse.ArgumentParser("Datasets converter from yolo to coco", add_help=True)
+    parser = argparse.ArgumentParser("Annotations converter from yolo to coco", add_help=True)
 
     parser.add_argument("--data_source", default="./coco128", help="Dataset root path")
     parser.add_argument("--class_names", default="./coco.name", help="Path of the label names")
@@ -21,7 +21,7 @@ def cli_main():
     args = parser.parse_args()
     print(f"Command Line Args: {args}")
 
-    converter = YOLO2COCO(args.data_source, args.class_names, split=args.split, year=args.year)
+    converter = AnnotationsConverter(args.data_source, args.class_names, split=args.split, year=args.year)
     converter.generate()
 
 

diff --git a/yolort/utils/yolo2coco.py → yolort/utils/annotations_converter.py b/yolort/utils/yolo2coco.py → yolort/utils/annotations_converter.py
@@ -8,9 +8,9 @@
 from PIL import Image
 
 
-class YOLO2COCO:
+class AnnotationsConverter:
     """
-    Convert YOLO labels to MSCOCO JSON labels.
+    Make a MSCOCO JSON format annotations from YOLO format.
 
     Args:
         root (string): Root directory of the Sintel Dataset