Spaces:
Runtime error
Runtime error
| 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 | |