commit for review

Author: shuichi

examples/bbox_to_visual_inspection_ai.py

@@ -0,0 +1,23 @@
+import json
+import os
+from visual_inspection_ai import VisualInspectionAi
+
+
+class VisualInspectionAiConverter:
+    def convert_from_bbox(self, file_path: str, export_file_path: str) -> dict:
+        bbox_json = self.open_bbox(file_path)
+        visual_inspection_ai = VisualInspectionAi.from_bbox(bbox_json)
+
+        os.makedirs(os.path.dirname(export_file_path), exist_ok=True)
+        with open(export_file_path, 'w') as file:
+            file.write(visual_inspection_ai.as_jsonl())
+
+    def open_bbox(self, file_path: str) -> any:
+        return json.load(open(file_path, "r"))
+
+
+converter = VisualInspectionAiConverter()
+
+file_path = "./import/visual_inspection_ai/annotations.json"  # 入力元のファイルパスを指定
+export_file_path = "./export/visual_inspection_ai/test.jsonl"  # 出力先のファイルパスを指定
+result = converter.convert_from_bbox(file_path, export_file_path)

examples/visual_inspection_ai.py

@@ -0,0 +1,47 @@
+import json
+
+
+# いろいろなフォーマットを受け取って、VisualInspectionAiのjsonlで出力するのが責務
+class VisualInspectionAi:
+
+    def __init__(self, vertex, project_type):
+        self.vertex = vertex
+        self.project_type = project_type
+
+    @classmethod
+    def from_bbox(cls, bbox_json):
+        # TODO bboxからvertexへの変換処理
+        vertex = [{"x": 0.1, "y": 0.1}]
+        return cls(vertex, "bbox")
+    
+    @classmethod
+    def from_segmentation(cls, segmentation_json, project_type):
+        # TODO segmentationからvertexへの変換処理
+        vertex = [{"x": 0.1, "y": 0.1}]
+        return cls(vertex, "segmentation")
+
+    # VisualInspectionAiのjsonlを作成する。
+    def __make_jsonl(self):
+        via = []
+        via_el = {
+             "image_gcs_uri": "gs://ffod-98sm/20240613/NG/test/BQ1334B__11-7_18-19_0004.png",
+             "vi_annotations": self.__make_vi_annotations_jsonl(),
+             "dataItemResourceLabels": {
+                 "goog_vi_ml_use": "test"
+                }
+        }
+        via.append(via_el)
+        return via
+
+    def __make_vi_annotations_jsonl(self):
+        return {
+            "viBoundingPoly": {
+                "vertex": self.vertex
+            },
+            "annotationSpec": "defect",
+            "annotationSet": "Polygons Regions" if self.project_type == "bbox" else "TODO segmentationの場合の表記調査"
+        }
+
+    def as_jsonl(self):
+        return json.dumps(self.__make_jsonl(), ensure_ascii=False, indent=4) + '\n'
+