MBA-DNet: A mask block attention-based foreign matter detection network for tobacco packages

Xiaobin Wang; Feixiang Shen; Chengjun Chen; Qiang Zhang

doi:10.1063/5.0185513

MBA-DNet: A mask block attention-based foreign matter detection network for tobacco packages

Rev Sci Instrum. 2024 Mar 1;95(3):035105. doi: 10.1063/5.0185513.

Authors

Xiaobin Wang¹, Feixiang Shen¹, Chengjun Chen¹, Qiang Zhang¹

Affiliation

¹ School of Mechanical and Automotive Engineering, Qingdao University of Technology, Qingdao, Shandong 266520, China.

PMID: 38441426
DOI: 10.1063/5.0185513

Abstract

Foreign matter, such as varia and mildew in the cutaway view of tobacco packages, can be detected using machine vision detection technology. However, mainstream object detection algorithms have poor detection ability for small targets when applied to foreign matter detection in the cutaway view of tobacco packages. To solve this problem, this study proposes Mask Block Attention (MBA) and introduces it into the feature extraction network to improve the global modeling ability of the object detection network, further enhancing its ability to detect foreign matter in the cutaway view of tobacco packages. Meanwhile, this study establishes a K-fold packet slicing defect dataset called K-PSDDS (K-fold packet slicing defect dataset) for foreign matter in the cutaway view of tobacco packages and conducts numerous experiments on K-PSDDS. The experimental results indicate that the AP50 and APbbox of DINO (DETR with an improved denoising anchor box for end-to-end target detection) + MBA reached 94.9% and 47.7%, respectively, showing an improvement of 0.3% and 0.9% over the baseline network DINO. Meanwhile, it achieves better performance and detection capabilities than fast region-based convolutional neural networks and other detection algorithms.