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.

Faster Detection Method of Driver Smoking Based on Decomposed YOLOv5

Research on Autonomous Driving Image Recognition Based on a New Real-Time Object Detection Model YOLOv5st

Research on Lightweight-Based Algorithm for Detecting Distracted Driving Behaviour

A smoking behavior detection method based on the YOLOv5 network

A Smoke Detection Model Based on Improved YOLOv5

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

Driver Abnormal Expression Detection Method Based on Improved Lightweight YOLOv5

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

An Improved YOLOv5s Fire Detection Model

Driver identification and fatigue detection algorithm based on deep learning

BiGA-YOLO: A Lightweight Object Detection Network Based on YOLOv5 for Autonomous Driving

YOLOFM: an improved fire and smoke object detection algorithm based on YOLOv5n

Tiny YOLO Optimization Oriented Bus Passenger Object Detection

A lightweight vehicles detection network model based on YOLOv5

Research on Fatigued-Driving Detection Method by Integrating Lightweight YOLOv5s and Facial 3D Keypoints

YOLO-SDH: improved YOLOv5 using scaled decoupled head for object detection

Smoking behavior detection algorithm based on YOLOv8-MNC

Making You Only Look Once Faster: Toward Real-Time Intelligent Transportation Detection

YOLOv4-5D: An Effective and Efficient Object Detector for Autonomous Driving

Smoking-YOLOv8: a novel smoking detection algorithm for chemical plant personnel

Fire-YOLO: A Small Target Object Detection Method for Fire Inspection