-
题名基于改进YOLOv3算法的连铸坯表面缺陷检测
被引量:4
- 1
-
-
作者
王睿泽
孙铁强
宋超
肖鹏程
-
机构
华北理工大学人工智能学院
华北理工大学河北省工业智能感知重点实验室
华北理工大学冶金与能源学院
-
出处
《组合机床与自动化加工技术》
北大核心
2023年第1期75-78,共4页
-
基金
国家自然科学基金(51904107)
河北省自然科学基金项目(E2020209005,E2021209094)。
-
文摘
针对连铸坯表面缺陷检测中存在的检测速度慢,检测精度低等问题,提出了一种改进的YOLOv3算法模型。使用轻量级网络MobileNetv2替换为YOLOv3的Darknet-53特征提取网络,优化网络结构,减小参数;使用Mate-ACON激活函数替换Leaky-ReLU函数,自适应的选择激活神经元;最后引入CIoU损失函数代替IoU边界框回归损失,提高模型定位的准确性,加速模型收敛。在连铸坯表面缺陷识别任务中,改进后的算法检测精度指标mAP达到96.96%,检测速度达到97 fps,满足实时性要求;模型训练得到的权重大小仅29 M,占用内存更低,能够满足连铸坯表面检测任务。
-
关键词
YOLOv3
缺陷检测
MobileNetv2
mate-acon
损失函数
-
Keywords
YOLOv3
defect detection
MobileNetv2
mate-acon
loss function
-
分类号
TH16
[机械工程—机械制造及自动化]
TG501
[金属学及工艺—金属切削加工及机床]
-