diff --git a/supervision/dataset/formats/yolo.py b/supervision/dataset/formats/yolo.py index fad8c8ae..39fb557b 100644 --- a/supervision/dataset/formats/yolo.py +++ b/supervision/dataset/formats/yolo.py @@ -98,7 +98,7 @@ def yolo_annotations_to_detections( relative_xyxy.append(polygon_to_xyxy(polygon=polygon)) if with_masks: relative_polygon.append(polygon) - + class_id = np.array(class_id, dtype=int) relative_xyxy = np.array(relative_xyxy, dtype=np.float32) relative_xyxyxyxy = np.array(relative_xyxyxyxy, dtype=np.float32)