节点文献

基于PP-PicoDet-XS的改进铝型材表面缺陷检测算法

Improved Surface Defects Detection Algorithm for Aluminum Profiles Based on PP-PicoDet-XS

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 马淑华李立振秦汉民沙晓鹏

【Author】 MA Shu-hua;LI Li-zhen;QIN Han-min;SHA Xiao-peng;School of Control Engineering,Northeastern University at Qinhuangdao;

【机构】 东北大学秦皇岛分校控制工程学院

【摘要】 铝型材在生产加工过程中会产生特征不明显和尺度大小不一等多类型的表面缺陷,针对现有人工抽检方法准确率低、实时性差、主观性强等问题,提出一种基于PP-PicoDet-XS的改进铝型材表面缺陷检测算法.改进的算法在主干网络中嵌入无参注意力SimAM,增强对深层有效特征的提取能力;使用SIoU(Scylla intersection over union)损失函数对训练过程进行优化,提高预测框的定位能力;采用量化蒸馏策略对模型进行压缩,提高推理速度.结果表明,改进的算法平均精度均值在交并比(intersection over union,IoU)阈值为0.5时达到了98.93%,在IoU阈值0.5~0.95范围内达到了57.60%,较未压缩的原始模型分别提高了1.73%和4.13%.将该算法部署到骁龙865移动端平台上进行推理,推理速度可达116.82帧/s,较未压缩的原始模型提高了47帧/s.

【Abstract】 During the production and processing of aluminum profiles, multiple types of surface defects such as unclear features and varying scales may generate. In response to the problems of low accuracy, poor real-time performance, and strong subjectivity in existing manual sampling method, an improved surface defects detection algorithm is proposed for aluminum profiles based on PP-PicoDet-XS. The SimAM attention was embedded in the backbone to enhance the ability of extracting deep effective features. The SIoU(Scylla intersection over union) loss function is used to optimize the training process to improve the positioning ability of the prediction boxes.The quantization and distillation were used to compress the model to improve the inference speed. The results show that the improved algorithm achieves a mean average precision of 98. 93% at intersection over union(IoU) threshold of 0. 5, and 57. 60% across IoU thresholds ranging from 0. 5 to 0. 95, which is 1. 73% and 4. 13% higher than the uncompressed original model. Deploying this algorithm on the Snapdragon 865 mobile platform for inference, the inference speed can reach 116. 82 frames per second, which is 47 frames per second higher than the uncompressed original model.

【基金】 河北省自然科学基金资助项目(F2021501021)
  • 【文献出处】 东北大学学报(自然科学版) ,Journal of Northeastern University(Natural Science) , 编辑部邮箱 ,2024年11期
  • 【分类号】TG146.21;TP391.41
  • 【下载频次】21
节点文献中: 

本文链接的文献网络图示:

本文的引文网络