Skip to content

Commit a0b7695

Browse files
author
shuichi
committed
change bbox convert logic
1 parent 6c3b8ac commit a0b7695

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

examples/visual_inspection_ai.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -80,13 +80,16 @@ def make_jsonl(self) -> dict:
8080
if self.project_type == "bbox"
8181
else self.__make_segmentation_vi_annotations_jsonl()
8282
)
83+
dataItemResourceLabels = {
84+
"goog_vi_ml_use": self.goog_vi_ml_use,
85+
}
86+
87+
if self.project_type == "segmentation":
88+
dataItemResourceLabels["label"] = self.label
8389
return {
8490
"image_gcs_uri": self.image_gcs_uri,
8591
"vi_annotations": vi_annotations,
86-
"dataItemResourceLabels": {
87-
"label": self.label,
88-
"goog_vi_ml_use": self.goog_vi_ml_use,
89-
},
92+
"dataItemResourceLabels": dataItemResourceLabels,
9093
}
9194

9295
def __make_bbox_vi_annotations_jsonl(self) -> dict:

0 commit comments

Comments
 (0)