一种交通场景中基于Mask R-CNN的车辆轴数识别方法

摘要

本发明公开了一种交通场景中基于Mask R‑CNN的车辆轴数识别方法，该方法以车轮识别为基础，其内容包括：车辆图像数据库的建立、Mask R‑CNN网络的训练、车辆与车轮掩膜的输出及车辆轴数的计算。在车辆轴数计算中，首先将图像中每个车轮掩膜中像素垂直坐标最小的点作为车轮掩膜的顶点，然后统计一个车辆掩膜中所包含的车轮掩膜顶点数量，该数量即为车辆轴数。本发明较基于车轮轮廓形状与模板匹配的识别方法精度更高，检测中只需布设一个交通监控摄像头，与传统采用埋入式传感器的车轴识别方式相比，所需设备成本低。