C#部署Yolov5步骤

1.导出ONNX模型

  1. 通过yolov5中的export.py导出ONNX模型。将export.py文件中的参数按训练的修改,如:‘–data’、‘–weights’、‘–imgsz’。点击运行。

2.C#调用yolo的

1. 下载C#运行YOLOv5的代码

1.github中有一个使用C#运行YOLOv5的代码:Yolov5Net

2. 安装程序包

  1. 安装如下第一个软件包。

软件包
需要安装以下库:
Microsoft.ML.OnnxRuntime.Gpu (ONNX运行时)
OpenCvSharp4.Extensions (输出图像框定对象使用)
OpenCvSharp4.Windows (输出图像框定对象使用)

3. 修改YoloCocoP6Model中的参数

  1. 我用yolov5-v6.1版本(yolov5-v5.0版本报超出索引,不知道为啥)。
  2. yolov5-v6.1版本需要用YoloCocoP6Model.cs,修改参数:
    1)将图片宽高(Width/Height)改为与模型训练/测试的相同。
    2)Dimensions为:标签种类+5。
    3)将Labels函数中的参数改为自己的标签。

参数修改

4.C#使用ONNX模型

  1. ONNX通过scorer调用,路径为绝对路径,使用的yolov5-v6.1版本。使用v5.0报错。
  2. 图片路径也为绝对路径。
  3. scorer = new YoloScorer(@“D:\skye\Yolov5Net-v1.0\Yolov5NetW\bin\best.onnx”, SessionOptions.MakeSessionOptionWithCudaProvider(0));—使用GPU检测。

在这里插入图片描述参考文章:
参考文章

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值