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

TP391.4; 铝型材在生产加工过程中会产生特征不明显和尺度大小不一等多类型的表面缺陷,针对现有人工抽检方法准确率低、实时性差、主观性强等问题,提出一种基于PP-PicoDet-XS的改进铝型材表面缺陷检测算法.改进的算法在主干网络中嵌入无参注意力SimAM,增强对深层有效特征的提取能力;使用SIoU(Scylla intersection over union)损失函数对训练过程进行优化,提高预测框的定位能力;采用量化蒸馏策略对模型进行压缩,提高推理速度.结果表明,改进的算法平均精度均值在交并比(intersection over union,IoU)阈值为0.5时达到了98.93%,在I...

Full description

Saved in:
Bibliographic Details
Published in东北大学学报（自然科学版） Vol. 45; no. 11; pp. 1557 - 1564
Main Authors 马淑华, 李立振, 秦汉民, 沙晓鹏
Format Journal Article
LanguageChinese
Published 东北大学秦皇岛分校 控制工程学院,河北 秦皇岛 066004 15.11.2024
Subjects
损失函数
SimAM
aluminum profiles
loss function
distillation
量化
蒸馏
defects detection
铝型材
缺陷检测
quantization
Online AccessGet full text
ISSN1005-3026
DOI10.12068/j.issn.1005-3026.2024.11.005

Cover

More Information
Summary:TP391.4; 铝型材在生产加工过程中会产生特征不明显和尺度大小不一等多类型的表面缺陷,针对现有人工抽检方法准确率低、实时性差、主观性强等问题,提出一种基于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.
ISSN:1005-3026
DOI:10.12068/j.issn.1005-3026.2024.11.005