Rename to AnnotationsConverter

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 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()
 
 

yolort/utils/__init__.py

@@ -9,22 +9,24 @@
 except ImportError:
     from torch.utils.model_zoo import load_url as load_state_dict_from_url
 
+from .annotations_converter import AnnotationsConverter
 from .dependency import check_version
 from .hooks import FeatureExtractor
 from .image_utils import cv2_imshow, get_image_from_url, read_image_to_tensor
 from .visualizer import Visualizer
 
 
 __all__ = [
+    "AnnotationsConverter",
+    "FeatureExtractor",
+    "Visualizer",
     "check_version",
     "contains_any_tensor",
     "cv2_imshow",
     "get_image_from_url",
     "get_callable_dict",
     "load_state_dict_from_url",
     "read_image_to_tensor",
-    "FeatureExtractor",
-    "Visualizer",
 ]
 
 

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