当前位置: 代码迷 >> 综合 >> 自己训练的yolov5模型用OpenVINO转换成IR模型后推理部署(win10+VS2017)
  详细解决方案

自己训练的yolov5模型用OpenVINO转换成IR模型后推理部署(win10+VS2017)

热度:59   发布时间:2023-12-10 02:02:29.0

作者:RayChiu_Labloy
版权声明:著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处


训练自己的yolov5模型

参考:labelimg制作VOC数据集并用yolov5训练目标检测模型_RayChiu757374816的博客-CSDN博客

VS2017和OpenVINO等的环境准备 

参考:yolov5预训练模型用OpenVINO转换成IR模型后推理部署(win10+VS2017)_RayChiu757374816的博客-CSDN博客

修改脚本 

修改label名称:

修改类别数目

        yolov5s.pt是基于80分类的,加上位置4个维度和一个置信度,一共85,我这里是2分类,那么是7,因此将脚本中两处85改为7即可。

修改IR文件和自己的输入图片以及device

还是修改155-157三行,对应自己的数据。

推理

【如果对您有帮助,交个朋友给个一键三连吧,您的肯定是我博客高质量维护的动力!!!】

  相关解决方案