Lightweight bus driver smoking detection

Xiaoqiang Wang; Xinyi Chen; Yanjie Wang

doi:10.1117/12.3055790

9 January 2025 Lightweight bus driver smoking detection

Xiaoqiang Wang, Xinyi Chen, Yanjie Wang

Author Affiliations +

Proceedings Volume 13486, Fourth International Conference on Computer Vision, Application, and Algorithm (CVAA 2024); 134862D (2025) https://doi.org/10.1117/12.3055790
Event: Fourth International Conference on Computer Vision, Application, and Algorithm (CVAA 2024), 2024, Chengdu, China

Abstract

The smoking behavior of bus drivers poses significant risks to the operational safety and service quality of public transport systems. To achieve high precision and lightweight detection of bus driver smoking, this paper presents a novel method that integrates YOLOv7-tiny with a channel pruning algorithm. Firstly, a small target detection layer is introduced atop the original three detection layers to enhance the model's capability in detecting cigarettes in small targets. Secondly, the standard 3×3 convolution is replaced by the Depth Separable Convolution (DSConv), thereby substituting complex computations with simpler linear computations, consequently reducing the computational load of the model. Thirdly, a combination of the Complete Intersection over Union (CIoU) and the Normalized Wasserstein Distance (NWD) loss functions is employed. This not only optimizes the convergence speed of network training but also enhances the accuracy of cigarette detection in bus drivers. Additionally, the channel pruning algorithm is applied to further lighten the model while maintaining or improving detection accuracy. Finally, ablation experiments are conducted on a self-constructed dataset of bus driver smoking to validate the effectiveness of the proposed method. Experimental results demonstrate that the improved model achieves a 3.3% enhancement in detection accuracy compared to the original YOLOv7-tiny algorithm, with a reduction of 73.3% in parameters and 66.9% in computation. These outcomes underscore the feasibility of deploying the model on mobile devices.

(2025) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Xiaoqiang Wang, Xinyi Chen, and Yanjie Wang "Lightweight bus driver smoking detection", Proc. SPIE 13486, Fourth International Conference on Computer Vision, Application, and Algorithm (CVAA 2024), 134862D (9 January 2025); https://doi.org/10.1117/12.3055790

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

;

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE