论文部分内容阅读
行人和车辆检测是计算机视觉中细分研究方向之一,广泛应用于视频监控、智能机器人等领域。针对城市场景中行人和车辆的检测问题,本文重点研究了卷积神经网络提取的多尺度特征利用和目标检测中边框(Bounding Box,BBox)回归,提出改进YOLOv3网络的行人和车辆检测方法。针对多尺度目标的检测问题,本文充分利用卷积神经网络(CNN)的多尺度特征研究了一种多尺度特征转换迁移融合的YOLOv3网络,以提高图像中多种尺度目标的检测准确度。首先,基于YOLOv3网络采用STDN算法的尺度转换模块对低层级特征图进行缩小,构造特征重用的主干网络提取特征。然后,采用尺度转换模块对高层级特征图进行放大,构建特征金字塔网络(Feature Pyramid Network,FPN)预测目标。在公开数据集MSCOCO上,该网络有效改善原YOLOv3网络因遮挡和目标较小导致的误检和漏检问题,各类别目标实验结果的精度均有提升。针对目标检测的定位和分类两大任务,本文提出基于置信度优化和级联BBox回归的检测模型。受two-stage目标检测模型Cascade R-CNN的启发,本文在one-stage目标检测模型基础上采用级联BBox回归方法。该方法通过设置递增的IOU(intersection over union)阈值,多阶段训练回归BBox,明显改善时间上不能匹配最佳的IOU值,检测的BBox偏差问题。同时,根据类间目标属性规律特征做置信度优化,以减少违背统计规律的误检。该方法基于现实场景中目标类别总是相互依存的这一先验规律,首先构造了相关系数矩阵和指数归一化的类间目标属性规律特征,然后利用这些特征在检测阶段优化目标的置信度打分。这一置信度优化方法利用客观存在的规律作为信息应用于目标检测,对目标检测效果有略微的提升。针对城市场景中行人和车辆检测问题,本文提出基于多尺度特征融合YOLOv3网络和级联BBox回归的行人和车辆检测模型。首先,考虑了遮挡、光线、场景等影响检测的因素,构建了有效的行人和车辆数据集,并采用类别平衡采样的方法解决了样本不平衡问题。然后,为加快训练模型收敛速度,降低漏检概率,采用K-means聚类优化方法选取先验框。最后,利用多尺度特征融合YOLOv3网络和级联BBox回归的模型对行人和车辆训练和检测。实验结果表明,本文基于改进YOLOv3网络的行人和车辆检测方法是可行且有效的,具有深远的参考价值。