File size: 201 Bytes
79c9d6f
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
from ultralytics import YOLO

model = YOLO('best.pt')

image_path = input("Enter image path: ")
results = model(image_path, device = 0, amp = 'True')  # return a list of Results objects

print(results)