YOLOv5缺陷检测工程文章目录

本文探讨了在工程实践中如何解决YOLOv5模型稳定性问题,重点关注提高检测速度的方法,如使用8bit模型、多进程/线程推理、batch处理以及将PyTorch模型转换为ONNX格式。作者提醒关注资源利用率以确定设备性能极限。
摘要由CSDN通过智能技术生成

这一块将会持续更新,一直到我手上的缺陷检测项目结束为止。但是这是一个小白的摸索过程,所以很多时候内容仍然浮在表面没办法深入。因此这里面的内容也会随着认知提高不断更新修正。

YOLOv5问题

YOLOv5无法稳定复现出一样结果的解决方案

无法稳定复现出一样结果的解决方案

工程方向提高YOLOv5检测速度

有一个问题你必须要注意。比如你现在使用的是yolov5x模型,并且工程中使用detect.py进行推理。现在你想提高检测速度。那么有一个很重要的点就是,在进行实验之前和实验中,你必须时刻关注你的CPU利用率、GPU利用率、内存利用率。当你的某一个已经爆掉了,其实就说明这是你这个设备的上限,并不一定是你的实验是完全错误的,也并不是方法没有效果。

半精度推理

YOLOv5半精度推理

使用8bit的模型检测8bit的图片

YOLOv5使用8bit的模型检测8bit的图片

使用detect.py进行多进程/线程推理

YOLOv5使用detect.py进行多进程推理
YOLOv5实现多线程推理

将图片合成batch进行推理

Yolov5 将图片合成batch进行推理-1
Yolov5 将图片合成batch进行推理-2

pt转为onnx

YOLOv5 pt转为onnx

YOLOv5 pt转engine

  • 41
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值