基于改进YOLOv5网络的轮胎规格字符识别  

Improving Tire Specification Character Recognition in YOLOv5 Network

作　　者：赵庆 魏鸿磊 杨祎宁 黄萌 ZHAO Qing;WEI Honglei;YANG Yining;HUANG Meng(School of Mechanical Engineering and Automation,Dalian Polytechnic University,Dalian 116034,China)

机构地区：[1]大连工业大学机械工程与自动化学院,大连116034

出　　处：《组合机床与自动化加工技术》2024年第2期164-168,共5页Modular Machine Tool & Automatic Manufacturing Technique

基　　金：辽宁省教育厅2021年度科学研究经费面上项目(LJKZ0535,LJKZ0526);2021年度本科教育教学综合改革项目(JGLX2021020,JCLX2021008)。

摘　　要：针对于汽车轮胎规格字符识别效率低、准确率低等问题,提出一种改进YOLOv5网络的轮胎规格字符识别方法。首先,将YOLOv5中的耦合头改为解耦头,提高网络的泛化能力;其次,提出C3-Faster模块,替换YOLOv5的Backbone和Head中的部分C3模块,提高网络的计算速度;最后,选用WIoU损失函数替换YOLOv5的CIoU损失函数,优化网络。通过对比实验,验证了C3-Faster和WIoU损失函数的有效性,在消融实验中,改进后的网络训练时间减少,mAP提高了3.7%,Precision提升2.1%。实验结果表明,该方法在汽车轮胎规格字符识别的有效性,提高了识别的准确性。A method for improving the recognition of characters indicating specifications for automobile tires is proposed to address the issues of low efficiency and accuracy in character recognition.This method involves improving the YOLOv5 network by replacing the coupled head with a decoupled head to enhance the network′s generalization ability.Additionally,a C3-Faster module is proposed to replace some C3 modules in the YOLOv5 backbone and head,thereby increasing the network′s computational speed.Finally,the WIoU loss function is selected to replace the CIoU loss function in YOLOv5 to optimize the network.Comparative experiments have been conducted to verify the effectiveness of the C3-Faster and WIoU loss function.The improved network reduced training time and increased mAP by 3.7% and precision by 2.1% in ablative experiments.The experimental results show that this method is effective in recognizing characters indicating specifications for automobile tires,improving recognition accuracy.

关 键 词：YOLOv5 解耦头 C3-Faster WIoU 轮胎规格字符识别 

分 类 号：TH161[机械工程—机械制造及自动化] TG66[金属学及工艺—金属切削加工及机床]