School of Control Engineering,Northeastern University at Qinhuangdao,Qinhuangdao 066004,China. Corresponding author: SHA Xiao-peng,E-mail: shaxiaopeng@neuq. edu. cn
In industrial products, various types of defects often exhibit high inter-class similarity, large scale variations, and complex backgrounds. To address these challenges, a lightweight industrial defect detection network (LIDD-Net) was proposed. To handle highly similar defect types, in LIDD-Net, a channel interaction separation backbone network was introduced, which enhanced feature extraction while reducing the computational cost of the model. To address multi-scale defect variations, a lightweight feature fusion network was developed, namely RepGhostPAN, to efficiently integrate multi-scale features in the image and accelerate inference. For complex detection backgrounds, a lightweight auxiliary training module was proposed, leveraging an auxiliary training head and a dynamic soft label assignment strategy to better distinguish target defects from complex backgrounds. Experiments on steel, aluminum, and tire defect datasets demonstrate that LIDD-Net achieves mAP@0.5 scores of 98.3%, 98.1%, and 96.1%, respectively, with only 0.62×106 parameters, meeting practical industrial requirements.
ShenX K, LiuJ H, JiangL, et al. A novel weld defect detection method for intelligent magnetic flux leakage detection system via contextual relation network[J]. IEEE Transactions on Industrial Electronics, 2024, 71(6): 6304-6314.
[2]
ZongZ F, SongG L, LiuY. DETRs with collaborative hybrid assignments training[C]// Proceedings of the IEEE Conference on International Conference on Computer Vision. Paris, 2023: 6725-6735.
[3]
YuZ P, HuangH B, ChenW J, et al. YOLO-FaceV2: a scale and occlusion aware face detector[J]. Pattern Recognition, 2024, 155: 110714.
[4]
DingX H, ZhangX Y, MaN N, et al. RepVGG: making VGG-style convnets great again[C] // Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Virtual, 2021: 13733-13742.
[5]
LeiY, ChenY R, HaiT, et al. DGNet: an adaptive lightweight defect detection model for new energy vehicle battery current collector[J]. IEEE Sensors Journal, 2023, 23: 29812-29830.
[6]
YeL Q. AugShuffleNet: communicate more, compute less [EB/OL]. (2022-03-13) [2022-08-21].
[7]
YangL X, ZhangR Y, LiL D, et al. SimAM: a simple, parameter-free attention module for convolutional neural networks[C]// International Conference on Machine Learning. Virtual, 2021: 11863-11874.
FilippasD, NicopoulosC, DimitrakopoulosG. Streaming dilated convolution engine[J]. IEEE Transactions on Very Large Scale Integration (VLSI) Systems, 2023, 31: 401-405.
[10]
ShamsolmoaliP, ChanussotJ, ZareapoorM, et al. Multipatch feature pyramid network for weakly supervised object detection in optical remote sensing images[J]. IEEE Transactions on Geoscience and Remote Sensing, 2022, 60: 1-13.
[11]
HanK, WangY H, TianQ, et al. Ghostnet: more features from cheap operations[C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Virtual, 2020: 1580-1589.
[12]
MishraR, GuptaH P. Designing and training of lightweight neural networks on edge devices using early halting in knowledge distillation[J]. IEEE Transactions on Mobile Computing, 2024, 23: 4665-4677.