💡💡💡深度学习工业缺陷检测
1)提供工业小缺陷检测性能提升方案,满足部署条件;
2)针对缺陷样品少等难点,引入无监督检测;
3)深度学习 C++、C#部署方案;
4)实战工业缺陷检测项目,学习如何选择合适的框架和模型;
1.基于Yolov8的工业小目标缺陷检测
1.工业油污数据集介绍
三星油污缺陷类别:头发丝和小黑点,["TFS","XZW"]
数据集大小:660张,包括部分良品图像,提升背景检测能力。
数据集地址:https://download.csdn.net/download/m0_63774211/87741209
缺陷特点:小目标缺陷,检测难度大,如下图所示;
1.1 动态蛇形卷积(Dynamic Snake Convolution),实现暴力涨点 | ICCV2023
Dynamic Snake Convolution | 亲测在工业小目标缺陷涨点明显,原始mAP@0.5 0.679提升至0.743
主要的挑战源于细长微弱的局部结构特征与复杂多变的全局形态特征。本文关注到管状结构细长连续的特点,并利用这一信息在神经网络以下三个阶段同时增强感知:特征提取、特征融合和损失约束。分别设计了动态蛇形卷积(Dynamic Snake Convolution),多视角特征融合策略与连续性拓扑约束损失。
我们希望卷积核一方面能够自由地贴合结构学习特征,另一方面能够在约束条件下不偏离目标结构太远。在观察管状结构的细长连续的特征后,脑海里想到了一个动物——蛇。我们希望卷积核能够像蛇一样动态地扭动,来贴合目标的结构。
基于Yolov8的工业小目标缺陷检测(2):动态蛇形卷积(Dynamic Snake Convolution),实现暴力涨点 | ICCV2023_AI小怪兽的博客-CSDN博客
1.2 微小目标检测可能存在检测能力不佳的现象,添加一个微小物体的检测头
多头检测器 | 亲测在工业小目标缺陷涨点明显,原始mAP@0.5 0.679提升至0.702
基于Yolov8的工业小目标缺陷检测(3):多检测头提升小目标检测精度_AI小怪兽的博客-CSDN博客
1.3 SPD-Conv,低分辨率图像和小物体涨点明显
SPD-Conv | 亲测在工业小目标缺陷涨点明显,原始mAP@0.5 0.679提升至0.775
SPD- conv由一个空间到深度(SPD)层和一个非跨步卷积层组成。SPD组件推广了一种(原始)图像转换技术[29]来对CNN内部和整个CNN的特征映射进行下采样:
基于Yolov8的工业小目标缺陷检测(4):SPD-Conv,低分辨率图像和小物体涨点明显_AI小怪兽的博客-CSDN博客
1.4大缺陷小缺陷一网打尽的轻量级目标检测器GiraffeDet
GiraffeDet | 亲测在工业小目标缺陷涨点明显,原始mAP@0.5 0.679提升至0.727
本文提出了GiraffeDet用于高效目标检测,giraffe包含轻量space-to-depth chain、Generalized-FPN以及预测网络
FPN旨在对CNN骨干网络提取的不同分辨率的多尺度特征进行融合。上图给出了FPN的进化,从最初的FPN到PANet再到BiFPN。我们注意到:这些FPN架构仅聚焦于特征融合,缺少了块内连接。因此,我们设计了一种新的路径融合GFPN:包含跳层与跨尺度连接,见上图d。
基于Yolov8的工业小目标缺陷检测(5):大缺陷小缺陷一网打尽的轻量级目标检测器GiraffeDet,暴力提升工业缺陷检测能力_AI小怪兽的博客-CSDN博客
1.5 多检测头结合小缺陷到大缺陷一网打尽的轻量级目标检测器GiraffeDet
多头检测器+ GiraffeDet | 亲测在工业小目标缺陷涨点明显,原始mAP@0.5 0.679提升至0.734
基于Yolov8的工业小目标缺陷检测(6):多检测头结合小缺陷到大缺陷一网打尽的轻量级目标检测器GiraffeDet,暴力提升工业小目标缺陷检测能力_AI小怪兽的博客-CSDN博客
1.6 Wasserstein Distance Loss,助力工业缺陷检测
Wasserstein Distance Loss | 亲测在工业小目标缺陷涨点明显,原始mAP@0.5 0.679提升至0.727
Wasserstein distance的主要优点是:
- 无论小目标之间有没有重叠都可以度量分布相似性;
- NWD对不同尺度的目标不敏感,更适合测量小目标之间的相似性。
NWD可应用于One-Stage和Multi-Stage Anchor-Based检测器。此外,NWD不仅可以替代标签分配中的IoU,还可以替代非最大抑制中的IoU(NMS)和回归损失函数。在一个新的TOD数据集AI-TOD上的大量实验表明,本文提出的NWD可以持续地提高所有检测器的检测性能。
基于Yolov8的工业小目标缺陷检测(7):Wasserstein Distance Loss,助力工业缺陷检测_AI小怪兽的博客-CSDN博客
未完待续,持续更新中
2.工业缺陷检测全流程解决方案
2.1 工业无监督缺陷检测,提升缺陷检测能力,解决缺陷样品少、不平衡等问题
工业无监督缺陷检测,提升缺陷检测能力,解决缺陷样品少、不平衡等问题_AI小怪兽的博客-CSDN博客
2.2 工业无监督缺陷检测,提升缺陷检测能力,解决缺陷样品少、不平衡等问题(二)
工业无监督缺陷检测,提升缺陷检测能力,解决缺陷样品少、不平衡等问题(二)_AI小怪兽的博客-CSDN博客
2.3 工业缺陷检测全流程解决方案,提供检测+后处理、分割+后处理、检测+RoI分割+后处理解决方案
工业缺陷检测全流程解决方案,提供检测+后处理、分割+后处理、检测+RoI分割+后处理解决方案_AI小怪兽的博客-CSDN博客
3.深度学习部署
3.1 Opencv DNN + C++部署
Yolov5 + Opencv DNN + C++部署_AI小怪兽的博客-CSDN博客
3.2 C++ GPU部署方式介绍:ONNX Runtime
Yolov5 C++ GPU部署方式介绍:ONNX Runtime_AI小怪兽的博客-CSDN博客
3.3 YOLOV8 Onnxruntime Opencv DNN C++部署
YOLOV8 Onnxruntime Opencv DNN C++部署_AI小怪兽的博客-CSDN博客
4.深度学习工业项目分享
4.1 基于yolov5的缺陷检测算法(工件缺陷)
基于yolov5的缺陷检测算法(工件缺陷)_yolov5做毕设的难点_AI小怪兽的博客-CSDN博客
4.2 手机背板缺陷分割
基于PaddleSeg的工业缺陷分割检测之手机背板缺陷分割_工业分割数据集_AI小怪兽的博客-CSDN博客
4.3 小目标摄像头镜头缺陷检测
基于Yolov5的工业缺陷检测之小目标摄像头镜头缺陷检测_yolov5缺陷检测_AI小怪兽的博客-CSDN博客
4.4 划痕缺陷检测
基于PaddleSeg的划痕检测_AI小怪兽的博客-CSDN博客
4.5 玻璃瓶盖缺陷检测
基于Yolov5的玻璃瓶盖缺陷检测_玻璃缺陷数据集_AI小怪兽的博客-CSDN博客
4.6 菌落(colony)计数
基于PaddleDetection的菌落(colony)计数_AI小怪兽的博客-CSDN博客