Format coco output data as per standard

This commit is contained in:
Mayank Agarwal 2023-07-23 15:15:09 +05:30
parent 7a07517750
commit bc0a38500d
1 changed files with 1 additions and 1 deletions

View File

@ -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],
"iscrowd": 0,
}
coco_annotations.append(coco_annotation)