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