from ultralytics import YOLO

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