基于轻量化YOLO的X射线焊缝图像信息检测

作者:谢经明; 刘默耘; 何文卓; 刘西策
来源:华中科技大学学报(自然科学版)科技大学, 2021, 49(01): 1-5.
DOI:10.13245/j.hust.210106

摘要

为了对X射线焊缝图像上包含许多信息的识别标记进行准确识别,提出了YOLO-G网络和YOLO-D网络.设计基于YOLO(you only look once)的X射线焊缝图像信息检测方法,并通过引入多种轻量化技术大幅降低了网络模型的参数量与计算量,在保证高检测精度的同时加快检测速度,降低对高算力硬件的依赖性.同时根据检测对象的特点对网络结构的预测层以及锚点框的尺寸和个数进行了适应性调整.最后在数据集上进行实验,结果显示:相比于YOLO-V3,YOLO-G与YOLO-D具有更高的准确率,网络模型分别降低了89.6%和79.68%,计算量仅为YOLO-V3的11.4%和21.3%.

全文