非限制场景下铁路机车车号定位检测方法 |
| |
引用本文: | 陈虎林,王焕民,米奡蔚. 非限制场景下铁路机车车号定位检测方法[J]. 铁路计算机应用, 2023, 32(4): 18-23. DOI: 10.3969/j.issn.1005-8451.2023.04.04 |
| |
作者姓名: | 陈虎林 王焕民 米奡蔚 |
| |
作者单位: | 1.中国铁路兰州局集团有限公司 科技和信息化部,兰州 730000 |
| |
基金项目: | 国家自然科学基金项目(61563027) |
| |
摘 要: | 针对传统铁路机车车号定位检测模型泛化性较低,不适用于多种检测应用场景等问题,提出一种适用于非限制场景、基于YOLO(You Only Look Once)v4-tiny模型的铁路机车车号定位检测方法。文章采用空洞卷积代替标准卷积,增大机车车号特征提取感受野,提升传统YOLOv4-tiny模型的检测精度;建立铁路机车车号数据集(RLND,Railway Locomotive Number Dataset),用于模型训练,并对模型的检测效果进行验证。验证结果表明,该方法对铁路机车车号的定位检测精度为99.44%,检测速度为50帧/s,能够应对非限制场景下的机车车号定位检测需求。
|
关 键 词: | 图像识别 机车车号定位 YOLOv4-tiny 非限制场景 轻量化 |
收稿时间: | 2022-10-14 |
Method for locating and detecting railway locomotive number in unrestricted scenarios |
| |
Affiliation: | 1.Department of Science, Technology and Information Technology, China Railway Lanzhou Group Co. Ltd., Lanzhou 730000, China2.Institute of Electronic Machinery Technology, Lanzhou Jiaotong University, Lanzhou 730070, China |
| |
Abstract: | In response to the low generalization of traditional railway locomotive number localization and detection models and their inability to adapt to various detection application scenarios, this paper proposed a method for locating and detecting railway locomotive number in unrestricted scenarios based on YOLO (You Only Look Once) v4-tiny model. The paper used cavity convolution instead of standard convolution to increase the receptive field of locomotive number feature extraction, improve the detection accuracy of the traditional YOLOv4 tiny model, established the Railway Locomotive Number Data set (RLND) for model training, and verified the detection effect of the model. The validation results show that the positioning and detection accuracy of this method for railway locomotive numbers is 99.44%, with a detection speed of 50 frames/s. It can meet the needs of locomotive number positioning and detection in unrestricted scenarios. |
| |
Keywords: | |
|
| 点击此处可从《铁路计算机应用》浏览原始摘要信息 |
|
点击此处可从《铁路计算机应用》下载免费的PDF全文 |
|