THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection

Home Page:https://arxiv.org/abs/2405.14458

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

如何用cpu推理onnx版本?

sunzx8 opened this issue · comments

commented

现在的代码是这样的
from ultralytics import YOLOv10
model = YOLOv10('/home/ubuntu/shawn/yolov10x.onnx')
results = model(source=image, conf=0.2, iou=0.8)[0]