C# 调用 Yolov8 模型

发布于 2024-07-04 124 次阅读

温馨提示：本文最后更新于2024年7月4日，已超过 30 天没有更新，某些文章具有时效性，若有错误或已失效，请在下方留言！

准备工作

环境：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");