Abstract:This paper discusses the application of deep learning technology in recognizing vehicle black smoke in road traffic monitoring videos. The use of massive surveillance video data imposes higher demands on the real-time performance of vehicle black smoke detection models. The YOLOv5s model, known for its excellent single-stage object detection performance, has a complex network structure. Therefore, this study proposes a lightweight real-time detection model for vehicle black smoke, named MGSNet, based on the YOLOv5s framework. The research involved collecting road traffic monitoring video data and creating a custom dataset for vehicle black smoke detection by applying data augmentation techniques such as changing image brightness and contrast. The experiment explored three different lightweight networks, namely ShuffleNetv2, MobileNetv3 and GhostNetv1, to reconstruct the CSPDarknet53 backbone feature extraction network of YOLOv5s. Comparative experimental results indicate that reconstructing the backbone network with MobileNetv3 achieved a better balance between detection accuracy and speed. The introduction of the squeeze excitation attention mechanism and inverted residual structure from MobileNetv3 effectively reduced the complexity of black smoke feature fusion. Simultaneously, a novel convolution module, GSConv, was introduced to enhance the expression capability of black smoke features in the neck network. The combination of depthwise separable convolution and standard convolution in the module further reduced the model's parameter count. After the improvement, the parameter count of the model is compressed to 1/6 of the YOLOv5s model. The lightweight vehicle black smoke real-time detection network, MGSNet, achieved a detection speed of 44.6 frames per second on the test set, an increase of 18.9 frames per second compared with the YOLOv5s model. The mAP@0.5 still exceeded 95%, meeting the application requirements for real-time and accurate detection of vehicle black smoke.

A High-Precision Fast Smoky Vehicle Detection Method Based on Improved Yolov5 Network

A Lightweight Model for Real-Time Detection of Vehicle Black Smoke

Cascaded Vehicle Matching and Short-Term Spatial-Temporal Network for Smoky Vehicle Detection

Accuracy Improvement of Automatic Smoky Diesel Vehicle Detection Using YOLO Model, Matching, and Refinement

A lightweight vehicles detection network model based on YOLOv5

Yolov5 Vehicle Detection Model in Fog Based on Channel Attention Enhancement

A Smoke Detection Model Based on Improved YOLOv5

A Lightweight Vehicle Detection Method Fusing GSConv and Coordinate Attention Mechanism

Lightweight Vehicle Detection Based on Improved YOLOv5s

SDV-Net: A Two-Stage Convolutional Neural Network for Smoky Diesel Vehicle Detection

A Vehicle Detection Method Based on an Improved U-YOLO Network for High-Resolution Remote-Sensing Images

Research on vehicle detection based on improved YOLOX_S.

A high-precision lightweight smoke detection model based on SE attention mechanism

Vehicle–Pedestrian Detection Method Based on Improved YOLOv8

An Improved YOLOv5s Fire Detection Model

Fast vehicle detection algorithm based on lightweight YOLO7-tiny

Lightweight Forest Smoke and Fire Detection Algorithm Based on Improved YOLOv5.

An Improved YOLO for Road and Vehicle Target Detection Model

Real-Time Vehicle Detection Based on Improved YOLO v5

Object Detection for Hazardous Material Vehicles Based on Improved YOLOv5 Algorithm

Road Traffic Vehicle Detection Method Using Lightweight YOLOv5 and Attention Mechanism