计算机与现代化 ›› 2023, Vol. 0 ›› Issue (08): 60-67.doi: 10.3969/j.issn.1006-2475.2023.08.010

• 人工智能 • 上一篇    下一篇

基于YOLOX结合DeepSort的船载车辆行人检测算法

  

  1. (江苏科技大学计算机学院,江苏 镇江 212100)
  • 出版日期:2023-08-30 发布日期:2023-09-13
  • 作者简介:刘昱杉(1997—),女,陕西宝鸡人,硕士研究生,研究方向:车船控制与智能数据处理,E-mail: 203070014@stu.just.edu.cn; 刘卫康(1998—),男,江苏连云港人,硕士研究生,研究方向:图像处理与智能交通,E-mail: 209070046@stu.just.edu.cn; 通信作者:刘庆华(1977—),男,内蒙古自治区赤峰人,教授,研究方向:车船电子与智能交通,E-mail: liuqh@just.edu.cn; 者甜甜(1997—),女,硕士研究生,研究方向:图像处理与机器视觉,E-mail: 202070028@stu.just.edu.cn; 王家晨(1998—),女,硕士研究生,研究方向:医疗图像与机器视觉,E-mail: 1518082637@qq.com。
  • 基金资助:
    国家自然科学基金资助项目(51008143); 江苏省六大高峰人才项目(XYDXX-117); 江苏省招商局海工科研创新基金资助项目(10901711-12); 苏州科技大学苏州智慧城市研究院开放基金资助项目(SZSCR2019011)

Pedestrian Detection Algorithm for Ship-borne Vehicles Based on YOLOX Combined#br# with DeepSort

  1. (School of Computer, Jiangsu University of Science and Technology, Zhenjiang 212100, China)
  • Online:2023-08-30 Published:2023-09-13

摘要: 摘要:针对目前国内轮渡车辆行人管控中缺乏对登船车辆行人进行实时捕捉与检测跟踪的问题,提出一种基于YOLOX改进的船载车辆及行人检测方法。首先,将高效通道注意力机制模块(ECA)加入原模型的加强特征提取网络的3个输出头处,提高网络对于车辆行人的特征提取能力;其次使用改进的ASPP模块替代原SPP模块,其中改进后的ASPP模块对原模块进行剪枝,并利用不同空洞卷积率的空洞卷积层相加解决原本ASPP模块的局部信息丢失问题;模型训练完成使用验证集进行验证后,与DeepSort结合进行跟踪检测并对登船车辆进行计数,从而判断轮渡是否超载。最后,本文所改进算法平均精准度指标(mAP)相比于原YOLOX算法提升了3.3%,精确率提升了4.4%,在GPU上测试运行速度达到55 FPS。实验结果表明,本文所改进算法适用于轮渡入口环境下对车辆、行人目标的实时性检测。

关键词: 关键词:轮渡, 目标检测, YOLOX, 深度可分离卷积, ASPP

Abstract: Abstract:Aiming at the lack of real-time capture, detection and tracking of boarding vehicles and pedestrians in the current domestic ferry vehicle pedestrian control, this paper proposes a ship-borne vehicle and pedestrian detection method based on improved YOLOX. Firstly, the enhanced channel attention module is added to the three output heads of the enhanced feature extraction network of the original model to improve the feature extraction capability of the network for vehicles and pedestrians. Secondly, we use the improved ASPP module to replace the original SPP module. Among them, the improved ASPP module prunes the original module, and uses the addition of atrous convolution layers with different atrous convolution rates to solve the problem of local information loss of the original ASPP module. After the model is trained and verified with the validation set, it is combined with DeepSort for tracking detection. Compared with the original YOLOX algorithm, the average accuracy index (mAP) of the improved algorithm in this paper is increased by 3.3%, the accuracy rate is increased by 4.4%, and the test running speed on the GPU reaches 55 FPS. The experimental results show that the improved algorithm in this paper is suitable for real-time detection of vehicles and pedestrians in the ferry entrance environment.

Key words: Key words: ferry service, target detection, YOLOX, depth separable convolution, ASPP

中图分类号: