Merge pull request #94 from fastlabel/develop

Merge to main

Author: ryoKaz

README.md

@@ -1043,10 +1043,10 @@ tasks = client.get_image_tasks(project="YOUR_PROJECT_SLUG")
 client.export_coco(tasks)
 ```
 
-Export with specifying output directory.
+Export with specifying output directory and file name.
 
 ```python
-client.export_coco(tasks=tasks, output_dir="YOUR_DIRECTROY")
+client.export_coco(tasks=tasks, output_dir="YOUR_DIRECTROY", output_file_name="YOUR_FILE_NAME")
 ```
 
 If you would like to export pose estimation type annotations, please pass annotations.
@@ -1055,7 +1055,7 @@ If you would like to export pose estimation type annotations, please pass annota
 project_slug = "YOUR_PROJECT_SLUG"
 tasks = client.get_image_tasks(project=project_slug)
 annotations = client.get_annotations(project=project_slug)
-client.export_coco(tasks=tasks, annotations=annotations, output_dir="YOUR_DIRECTROY")
+client.export_coco(tasks=tasks, annotations=annotations, output_dir="YOUR_DIRECTROY", output_file_name="YOUR_FILE_NAME")
 ```
 
 ### YOLO

fastlabel/__init__.py

@@ -1082,18 +1082,22 @@ def __get_yolo_format_annotations(self, dataset_folder_path: str) -> dict:
 
     # Task Convert
 
-    def export_coco(self, tasks: list, annotations: list = [], output_dir: str = os.path.join("output", "coco")) -> None:
+    def export_coco(self, tasks: list, annotations: list = [], output_dir: str = os.path.join("output", "coco"), output_file_name: str = "annotations.json") -> None:
         """
         Convert tasks to COCO format and export as a file.
         If you pass annotations, you can export Pose Estimation type annotations.
 
         tasks is a list of tasks. (Required)
         annotations is a list of annotations. (Optional)
         output_dir is output directory(default: output/coco). (Optional)
+        output_file_name is output file name(default: annotations.json). (Optional)
         """
+        if not utils.is_json_ext(output_file_name):
+            raise FastLabelInvalidException(
+                "Output file name must have a json extension", 422)
         coco = converters.to_coco(tasks, annotations)
         os.makedirs(output_dir, exist_ok=True)
-        file_path = os.path.join(output_dir, "annotations.json")
+        file_path = os.path.join(output_dir, output_file_name)
         with open(file_path, 'w') as f:
             json.dump(coco, f, indent=4, ensure_ascii=False)
 

fastlabel/utils.py

@@ -19,6 +19,10 @@ def is_video_supported_ext(file_path: str) -> bool:
     return file_path.lower().endswith('.mp4')
 
 
+def is_json_ext(file_name: str) -> bool:
+    return file_name.lower().endswith('.json')
+
+
 def get_basename(file_path: str) -> str:
     """
     e.g.) file.jpg -> file

setup.py

@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name="fastlabel",
-    version="0.11.11",
+    version="0.11.12",
     author="eisuke-ueta",
     author_email="[email protected]",
     description="The official Python SDK for FastLabel API, the Data Platform for AI",