基于YOLO v3算法的不同主干网络对织物瑕疵检测

• 作者: 谢景洋 王巍 刘婷
• 来源期刊: 测控技术
• 关键词: YOLOv3 主干网络替换 织物瑕疵检测 Grad-CAM
• 年，卷（期）: 2021,（3）
• 所属期刊栏目: 模式识别与人工智能
• 页码范围: 61-66,95
• 页数: 7页
• 分类号: TP391
• 语种: 中文
• DOI: 10.19708/j.ckjs.2021.03.012

摘要

针对布匹瑕疵检测中由于输入图片尺寸较大导致检测速度降低的问题,提出了使用轻量化的Mo-bileNets、更细粒度多尺度特征的Res2 Net修改YOLO v3主干网络的两种方案.在相同条件下训练3种YOLO v3网络,实现了对20种缺陷的检测.对比搭载3种主干网络YOLO v3算法的参数量、检测速度和精度,利用可视化工具Grad-CAM研究3种主干网络对缺陷的特征提取能力.实验结果表明,搭载MobileNets主干网络的参数量是原始网络的1/3,检测速度提高30％,并且在精度上mAP值达到31.22％,接近原始网络的32.54％.Res2 Net的细粒度多尺度特征结构在布匹瑕疵检测上没有明显优势.Grad-CAM可视化结果表明MobileNets主干网络在特征提取上更专一.