.png){kind=icon}
返回
DFP-YOLO: a lightweight machine tool workpiece defect detection algorithm based on computer vision
.png){kind=icon}
PRE
AI
DOI：10.1007/s00371-024-03705-8.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Achieving accurate identification of various defects in workpiece surface defect detection is a complex task. To address the issues of high computational complexity and model complexity in existing defect detection algorithms, this paper proposes an improved network model algorithm, DFP-YOLO, for defect detection on cutting tool workpieces. The improved algorithm first introduces the C3_D module and C3_F module to reduce model parameters and complexity. Secondly, the bounding box regression loss function utilizes SIoU to improve detection accuracy. To address the issues of imbalanced detection samples and long training time, transfer learning is employed to accelerate the convergence speed of the overall network model and save computational resources. Finally, channel pruning algorithm is applied to compress the overall network structure without significantly affecting detection accuracy, resulting in a more lightweight model. The experimental results demonstrate that DFP-YOLO effectively reduces the parameter quantity by 51% and GFLOPs by 50% while preserving the mean average precision (mAP) unchanged. Moreover, the detection speed remains constant at 108 frames per second. This improved algorithm surpasses existing methods for surface defect detection and achieves both fast and accurate detection of defects on workpieces. Code is available at https://github.com/wdikky/DFP-YOLO
Keyword:
YOLOv5
Transfer learning
Pruning
Lightweight
Computer vision
期刊
IF:
2.9
论文数:
4.6K
被引数:
6.5K
机构
引用论文
Generalizable end-to-end deep learning frameworks for real-time attitude estimation 6DoF inertial measurement units.png)
MEASUREMENT
OAAI
MEASUREMENTIF5.6
OAAI分享收藏RDD-YOLO: A modified YOLO for detection of steel surface defectsRdd-yolo: 一种用于检测钢材表面缺陷的改进YOLOMEASUREMENT
MEASUREMENTIF5.6
PREAI
分享收藏An End-to-End Steel Surface Defect Detection Approach via Fusing Multiple Hierarchical Features融合多个层次特征的端到端钢表面缺陷检测方法OAAI
OAAI分享收藏Automatic Detection and Classification System of Domestic Waste via Multimodel Cascaded Convolutional Neural Network基于多模型级联卷积神经网络的生活垃圾自动检测分类系统OAAI
OAAI分享收藏Benchmark Revision for HOG-SVM Pedestrian Detector Through Reinvigorated Training and Evaluation Methodologies通过重新激活的培训和评估方法对hog-svm行人检测器进行基准修订
PREAI
分享收藏分享收藏
查看更多内容