Spaces:

Dirake
/

Pokemon2

Runtime error

Pokemon2 / predict.py

Upload 13 files

acbc0af verified about 1 year ago

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)   

	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)