Suppr超能文献

改进的YOLOV7-SSWD数字读数识别算法研究

Research on improved YOLOV7-SSWD digital meter reading recognition algorithms.

作者信息

Cao Zhenguan, Yang Haixia, Fang Liao, Li Zhuoqin, Li Jinbiao, Dong Gaohui

机构信息

School of Electrical and Information Engineering, Anhui University of Science and Technology, Huainan 232001, Anhui, China.

出版信息

Rev Sci Instrum. 2024 Sep 1;95(9). doi: 10.1063/5.0207733.

Abstract

Meter reading recognition is an important link for robots to complete inspection tasks. To solve the problems of low detection accuracy and inaccurate localization of current meter reading recognition algorithms, the YOLOV7-SSWD (YOLOV7-SiLU-SimAM-Wise-IoU-DyHeads) model is proposed, a novel detection model based on the multi-head attention mechanism, which is improved on the YOLOV7-Tiny model. First, the Wise-IoU loss function is used to solve the problem of sample quality imbalance and improve the model's detection accuracy. Second, a new convolutional block is constructed using the SiLU activation function and applied to the YOLOV7-Tiny model to enhance the model's generalization ability. The dynamic detection header is then built as the header of YOLOV7-Tiny, which realizes the fusion of multi-scale feature information and improves the target recognition performance. Finally, we introduce SimAM to improve the feature extraction capability of the network. In this paper, the importance of each component is fully verified by ablation experiments and comparative analysis. The experiments showed that the mAP and F1-scores of the YOLOV7-SSWD model reached 89.8% and 0.84. Compared with the original network, the mAP increased by 8.1% and the F1-scores increased by 0.1. The YOLOV7-SSWD algorithm has better localization and recognition accuracy and provides a reference for deploying inspection robots to perform automatic inspections.

摘要

抄表识别是机器人完成巡检任务的重要环节。针对当前抄表识别算法检测精度低、定位不准确的问题,提出了YOLOV7-SSWD(YOLOV7-SiLU-SimAM-Wise-IoU-DyHeads)模型,这是一种基于多头注意力机制的新型检测模型,在YOLOV7-Tiny模型基础上进行了改进。首先,使用Wise-IoU损失函数解决样本质量不平衡问题,提高模型检测精度。其次,利用SiLU激活函数构建新的卷积块并应用于YOLOV7-Tiny模型,增强模型泛化能力。然后构建动态检测头作为YOLOV7-Tiny的头部,实现多尺度特征信息融合,提高目标识别性能。最后,引入SimAM提高网络特征提取能力。本文通过消融实验和对比分析充分验证了各组件的重要性。实验表明,YOLOV7-SSWD模型的mAP和F1分数分别达到89.8%和0.84。与原网络相比,mAP提高了8.1%,F1分数提高了0.1。YOLOV7-SSWD算法具有更好的定位和识别精度,为部署巡检机器人进行自动巡检提供了参考。

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验