将yolov8权重文件转为onnx格式并在c#中使用

yolo模型转ONNX

在yolov8中，我们将训练结果的.pt权重文件转换为onnx格式只需要使用ultralytics库中的YOLO类，使用pip安装ultralytics库，然后执行下面python代码

from ultralytics import YOLO

# 加载YOLOv8模型
model = YOLO("best.pt")

# 将模型导出为ONNX格式
success = model.export(format="onnx", simplify=True)

# 检查是否成功导出
assert success

 此程序就会加载当前根目录下的best.pt并转换为ONNX格式进行导出

 C#中使用ONNX

创建一个.NET 8.0的项目，因为自己写比较繁琐，我们直接使用封装好的Nuget包YoloDotNet

地址：NuGet Gallery | Yol