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

Merge branch 'OBB' of https://github.com/Bhavay-2001/roboflow-supervision into OBB

This commit is contained in:
Bhavay Malhotra 2024-06-06 22:14:11 +05:30
parent 6ca09ccbfc 859f0258bc
commit 1ea9345845
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
supervision/dataset/formats/yolo.py
View File

@ -93,7 +93,7 @@ def yolo_annotations_to_detections(
else:
relative_xyxy.append(polygon_to_xyxy(polygon=polygon))
if with_masks:
relative_polygon.append(polygon)
relative_polygon.append(polygon)
class_id = np.array(class_id, dtype=int)
relative_xyxy = np.array(relative_xyxy, dtype=np.float32)
@ -112,7 +112,11 @@ def yolo_annotations_to_detections(
(polygon * np.array(resolution_wh)).astype(int) for polygon in relative_polygon
]
mask = _polygons_to_masks(polygons=polygons, resolution_wh=resolution_wh)
return Detections(class_id=class_id, xyxy=xyxy, data=data, mask=mask) if is_obb else Detections(class_id=class_id, xyxy=xyxy, mask=mask)
return (
Detections(class_id=class_id, xyxy=xyxy, data=data, mask=mask)
if is_obb
else Detections(class_id=class_id, xyxy=xyxy, mask=mask)
)
def load_yolo_annotations(