File size: 389 Bytes
89d4d16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import ultralytics
from ultralytics import YOLO
model = YOLO('yolov8n.pt')
def detected_objects(filename:str):
results = model.predict(source=filename, conf=0.25)
categories = results[0].names
dc = []
for i in range(len(results[0])):
cat = results[0].boxes[i].cls
dc.append(categories[int(cat)])
print(dc)
return results, dc
|