科研之友
微信
新浪微博
Facebook
分享链接
摘要
为解决人工查看4C装置拍摄的接触网图片速度慢、效率低的问题,提出一种基于改进YOLOv4-tiny的铁路接触网主要部件的检测方法。针对接触网图片分辨率大、部件小的情况,采用预处理方法裁减掉冗余黑色背景,提高部件的相对大小;增加一个对于小目标友好的下采样8倍的输出特征图;为避免一些小部件在主干网络提取特征时的语义信息丢失和占比减少问题,将原网络残差结构中的3*3卷积改进为并列的1*1、3*3和5*5卷积。实验结果表明,改进后的算法在准确率上相比YOLOv4-tiny提高了32.11%,达到96.56%。
- 单位
