File size: 194 Bytes
cb8ea16
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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)