Pokemon / predict.py
Dirake's picture
Create predict.py
cb8ea16 verified
raw
history blame contribute delete
194 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)