30 June 2021 Seatbelt detection in road surveillance images based on improved dense residual network with two-level attention mechanism
Jingrui Luo, Jinbo Lu, Guangde Yue
Author Affiliations +
Abstract

Seatbelt detection is an important topic in intelligent transport systems. The accuracy of seatbelt detection in traffic video surveillance is affected by many factors, such as complex road environments, lighting, weather, direction of the camera, etc., which bring difficulties for traditional image processing methods. We propose a seatbelt detection method based on classification using convolutional neural networks. Dense residual blocks are used to avoid gradient dispersion and information loss during the network training. A two-level attention mechanism is also introduced to further improve the network performance by simultaneously modeling the correlation among different feature channels using channel attention mechanism, and correlation among different pixels within the feature map using pixel attention mechanism. To make the whole system complete, we added the driver area localization module, which is accomplished using the YOLOV3 model. Small images concentrated on the driver location, which are obtained through the localization operation from larger images, are passed into the classification network for seatbelt detection. We compared the proposed method with several other methods. Comparative results show that the proposed method has higher accuracy and is more robust for seatbelt detection for complex environments.

© 2021 SPIE and IS&T 1017-9909/2021/$28.00© 2021 SPIE and IS&T
Jingrui Luo, Jinbo Lu, and Guangde Yue "Seatbelt detection in road surveillance images based on improved dense residual network with two-level attention mechanism," Journal of Electronic Imaging 30(3), 033036 (30 June 2021). https://doi.org/10.1117/1.JEI.30.3.033036
Received: 7 March 2021; Accepted: 15 June 2021; Published: 30 June 2021
Lens.org Logo
CITATIONS
Cited by 2 scholarly publications and 1 patent.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Roads

Image classification

Surveillance

Feature extraction

Lutetium

Convolution

Light sources and illumination

Back to Top