首页|安检图像小目标违禁品特征提取模块构建与应用

安检图像小目标违禁品特征提取模块构建与应用

扫码查看
针对物流包裹安检图像中小目标违禁品易漏检问题,通过在感受野模块的多分支并行网络上引入卷积注意力模块,构建一种适用于小目标违禁品检测的特征提取模块。在此基础上,将构建的特征提取模块融入YOLOv5 模型的主干部分,使得模型在违禁品检测的过程中聚焦于图像的重要特征。为了充分发挥所构建模块对于小目标物体的特征提取能力,采用空间深度转换模块替代原模型中的下采样模块,使得YOLOv5 模型在特征提取的过程中能够尽可能地保留小目标物体的特征信息,提高对小目标违禁品的检测效果。
Construction and Application of a Feature Extraction Module for Small Target Prohibited Items in Security Inspection Images
Aiming at the problem that small target prohibited items in logistics package security inspection images is easy to miss detection,a feature extraction module suitable for small target prohibited items detection is constructed by introducing a convolutional attention module on the multi-branch parallel network of the receptive field module.On this basis,the constructed feature extraction module is integrated into the backbone of the YOLOv5 model,so that the model focuses on the important features of the image in the process of prohibited items detection.In order to fully utilize the feature extraction ability of the constructed module for small target objects,a spatial depth conversion module is used to replace the downsampling module in the original model,so that the YOLOv5 model can retain the feature information of small target objects as much as possible during the feature extraction process,and improve the detection effect for small target prohibited items.

security inspection imagesmall target prohibited itemfeature extraction modulecomputer visionlogistics

刘天时、周泽华、郝敏杰

展开 >

西安石油大学 计算机学院,陕西 西安 710065

安检图像 小目标违禁品 特征提取模块 计算机视觉 物流包裹

2024

现代信息科技
广东省电子学会

现代信息科技

ISSN:2096-4706
年,卷(期):2024.8(4)
  • 12