C# 调用 Yolov8 模型

发布于 12 天前  28 次阅读


准备工作

环境:VS2022,.NET 8

需要安装的Nuget包:YoloV8.Gpu,Microsoft.ML.OnnxRuntime.Gpu

步骤

1. 将训练好的pt文件转化为onnx

from ultralytics import YOLO

# Load a model
model = YOLO('path/to/best')

# export the model to ONNX format
model.export(format='onnx')

2. 在C#中推理

using Compunet.YoloV8;
using Compunet.YoloV8.Plotting;
using SixLabors.ImageSharp;

var image = Image.Load("path/to/image");
var predictor = YoloV8Predictor.Create("path/to/model");
var result = await predictor.DetectAsync(image);
var plotted = await result.PlotImageAsync(image);
plotted.Save("./pose_demo.jpg");
最后更新于 2024-07-04