File size: 159 Bytes
d025638
 
9f24354
2aba432
1
2
3
4
5
from ultralytics import YOLO

model = YOLO('yolov8m-seg.pt')
prediction = model.predict('trash_01.jpg', imgsz=(1024,1024), show=True, save=True, save_dir=".")