gun_dtct / app.py
anonymousatom's picture
Update app.py
79c9d6f verified
raw
history blame contribute delete
201 Bytes
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)