File size: 501 Bytes
6add590 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from ultralytics import YOLO
def detect_bubbles(model_path, image_path):
"""
Detects bubbles in an image using a YOLOv8 model.
Args:
model_path (str): The file path to the YOLO model.
image_path (str): The file path to the input image.
Returns:
list: A list containing the coordinates, score and class_id of
the detected bubbles.
"""
model = YOLO(model_path)
bubbles = model(image_path)[0]
return bubbles.boxes.data.tolist()
|