xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

accomodate for no polygon indices

This commit is contained in:
Mayank Agarwal 2023-07-23 20:52:07 +05:30
parent bc0a38500d
commit 6587f39b0f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
supervision/dataset/formats/coco.py
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] if polygon else [],
"iscrowd": 0,
}
coco_annotations.append(coco_annotation)