YOLO lightweight contraband detection network using attention mechanism

Yifei Dai; Puchun Chen

doi:10.1117/12.2672161

28 February 2023 YOLO lightweight contraband detection network using attention mechanism

Yifei Dai, Puchun Chen

Proceedings Volume 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022); 1259618 (2023) https://doi.org/10.1117/12.2672161
Event: International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 2022, Changsha, China

Abstract

In stations, airports and other places, contraband detection faces many problems such as false positives, omissions and slow detection speed caused by object background interference and human factors. This paper proposes an improved network based on YOLO-lightweight. The attention mechanism module is embedded in the backbone network, focusing on the important features from different channels. CBAM-FPN (Convolution Block Attention Module and Feature Pyramid Networks) structure is adopted in the network neck to reduce the loss of network features. Attention mechanism module is added in the bottom-up feature fusion process. Finally, CIOU is used as the edge optimization loss function to accelerate the network convergence and optimize the network model. Compared with YOLOv4-tiny, the precision is improved by 3.8%, reaching 87.5%. The detection speed reaches 60.3fps. The improved network only occupies 23.4M memory, which is convenient for embedding mobile devices. The improved network meets the real-time detection requirements.

Citation Download Citation

Yifei Dai and Puchun Chen "YOLO lightweight contraband detection network using attention mechanism", Proc. SPIE 12596, International Conference on Mechatronics Engineering and Artificial Intelligence (MEAI 2022), 1259618 (28 February 2023); https://doi.org/10.1117/12.2672161

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

Includes PDF, HTML & Video, when available