pemix09's picture
init
78db68c verified
raw
history blame contribute delete
278 Bytes
from ultralytics import YOLO
import cv2
model_path = 'runs/segment/train21/weights/best.pt'
image_path = '/Users/mac/Desktop/Screenshot 2024-06-08 at 15.12.26.png'
model = YOLO(model_path)
results = model.predict(image_path, save_conf=True, show_boxes=False, save=True)