作者: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三行,对应自己的数据。
推理
【如果对您有帮助,交个朋友给个一键三连吧,您的肯定是我博客高质量维护的动力!!!】