改进FCOS算法的车辆检测方法研究

IMPROVED FCOS ALGORITHM FOR VEHICLE DETECTION

  • 摘要: 针对目前车辆检测的误差率高、检测速度慢等问题,提出一种基于改进全卷积单阶段(Fully Convolutional One-Stage Object Detection,FCOS)的车辆检测算法。通过引入一种考虑多个几何特征的交并比损失函数,改善了训练过程中高长宽比车辆、并行车辆难以准确回归的现象;使用多尺度卷积结合多维特征信息,增强了算法对不同尺度检测的鲁棒性;根据车辆检测场景改进了回归尺度,提高模型的推理准确度。实验结果表明,该方法在车辆检测任务中能够明显提升检测精度并保持检测速度不下降。

     

    Abstract: Aimed at the problems of high error rate and slow detection speed in vehicle detection, an improved fully convolutional one-stage object detection vehicle detection method is proposed. An intersection and union ratio loss function considering multiple geometric factors was introduced, which improved the phenomenon that it was difficult for high aspect ratio vehicles and parallel vehicles to regress accurately in the training process. Multiscale convolution was used to combine multi-dimensional features information, and the robustness of the algorithm to different scale detection was enhanced. According to the scene of vehicle detection, the regression scale was improved to improve the reasoning accuracy of the model. The experimental results show that this method can significantly improve the detection accuracy while maintaining the detection speed in vehicle detection tasks.

     

/

返回文章
返回