Pokemon2 / predict.py
Dirake's picture
Upload 13 files
acbc0af verified
raw
history blame contribute delete
195 Bytes
import ultralytics
from ultralytics import YOLO
ultralytics.checks()
import torch
device = 'cuda'
model = YOLO("best.pt")
path ="test.mp4"
results = model.predict(source=path, show = True)