🛠️ fix coco segmentation data for polygon should be [polygon,] not polygon
Format coco output data as per standard
This commit is contained in:
commit
8662671e6b
|
|
@ -124,7 +124,7 @@ def detections_to_coco_annotations(
|
|||
"category_id": int(class_id),
|
||||
"bbox": [xyxy[0], xyxy[1], box_width, box_height],
|
||||
"area": box_width * box_height,
|
||||
"segmentation": polygon,
|
||||
"segmentation": [polygon] if polygon else [],
|
||||
"iscrowd": 0,
|
||||
}
|
||||
coco_annotations.append(coco_annotation)
|
||||
|
|
|
|||
Loading…
Reference in New Issue