Light "You Only Look Once": An Improved Lightweight Vehicle-Detection Model for Intelligent Vehicles under Dark Conditions

被引:1
|
作者
Yin, Tianrui [1 ]
Chen, Wei [1 ]
Liu, Bo [2 ]
Li, Changzhen [3 ]
Du, Luyao [1 ]
机构
[1] Wuhan Univ Technol, Sch Automat, Wuhan 430070, Peoples R China
[2] Wuhan Zhongyuan Elect Grp Co Ltd, Wuhan 430010, Peoples R China
[3] Wuhan Univ Technol, Sch Informat Engn, Wuhan 430070, Peoples R China
关键词
intelligent vehicles; vehicle detection; lightweight network; feature pyramid network;
D O I
10.3390/math12010124
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Vehicle detection is crucial for traffic surveillance and assisted driving. To overcome the loss of efficiency, accuracy, and stability in low-light conditions, we propose a lightweight "You Only Look Once" (YOLO) detection model. A polarized self-attention-enhanced aggregation feature pyramid network is used to improve feature extraction and fusion in low-light scenarios, and enhanced "Swift" spatial pyramid pooling is used to reduce model parameters and enhance real-time nighttime detection. To address imbalanced low-light samples, we integrate an anchor mechanism with a focal loss to improve network stability and accuracy. Ablation experiments show the superior accuracy and real-time performance of our Light-YOLO model. Compared with EfficientNetv2-YOLOv5, Light-YOLO boosts mAP@0.5 and mAP@0.5:0.95 by 4.03 and 2.36%, respectively, cuts parameters by 44.37%, and increases recognition speed by 20.42%. Light-YOLO competes effectively with advanced lightweight networks and offers a solution for efficient nighttime vehicle-detection.
引用
收藏
页数:19
相关论文
共 46 条
  • [1] An Improved You Only Look Once Based Intelligent System for Moving Vehicle Detection
    Ghosh, Rajib
    INTERNATIONAL JOURNAL OF INTELLIGENT TRANSPORTATION SYSTEMS RESEARCH, 2023, 21 (02) : 310 - 318
  • [2] An Improved You Only Look Once Based Intelligent System for Moving Vehicle Detection
    Rajib Ghosh
    International Journal of Intelligent Transportation Systems Research, 2023, 21 : 310 - 318
  • [3] Robust Vehicle Detection Based on Improved You Look Only Once
    Kumar, Sunil
    Jailia, Manisha
    Varshney, Sudeep
    Pathak, Nitish
    Urooj, Shabana
    Abd Elmunim, Nouf
    CMC-COMPUTERS MATERIALS & CONTINUA, 2023, 74 (02): : 3561 - 3577
  • [4] A rail defect detection framework under class-imbalanced conditions based on improved you only look once network
    Ding, Yu
    Zhao, Qin
    Li, Tianhao
    Lu, Chen
    Tao, Laifa
    Ma, Jian
    Engineering Applications of Artificial Intelligence, 2024, 138
  • [5] Lightweight You Only Look Once v8: An Upgraded You Only Look Once v8 Algorithm for Small Object Identification in Unmanned Aerial Vehicle Images
    Huangfu, Zhongmin
    Li, Shuqing
    APPLIED SCIENCES-BASEL, 2023, 13 (22):
  • [6] Automatic detection of sewer defects based on improved you only look once algorithm
    Tan, Yi
    Cai, Ruying
    Li, Jingru
    Chen, Penglu
    Wang, Mingzhu
    AUTOMATION IN CONSTRUCTION, 2021, 131
  • [7] Improved lightweight underwater target detection network based on YOLOV4 (you only look once v4)
    Shi X.
    Wang H.
    Harbin Gongcheng Daxue Xuebao/Journal of Harbin Engineering University, 2023, 44 (01): : 154 - 160
  • [8] A lightweight model based on you only look once for pomegranate before fruit thinning in complex environment
    Du, Yurong
    Han, Youpan
    Su, Yaoheng
    Wang, Jiuxin
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 137
  • [9] Passenger Flow Detection in Subway Stations Based on Improved You Only Look Once Algorithm
    Li, Xianwang
    Zhang, Yuxiang
    He, Deqiang
    Teng, Xiaoliang
    Liu, Bin
    Chen, Yanjun
    TRANSPORTATION RESEARCH RECORD, 2023, 2677 (09) : 397 - 409
  • [10] Making You Only Look Once Faster: Toward Real-Time Intelligent Transportation Detection
    Dai, Yuan
    Liu, Weiming
    Xie, Wei
    Liu, Ruikang
    Zheng, Zhongxing
    Long, Kejun
    Wang, Liang
    Mao, Liang
    Qiu, Qisheng
    Ling, Guangzheng
    IEEE INTELLIGENT TRANSPORTATION SYSTEMS MAGAZINE, 2023, 15 (03) : 8 - 25