首页|基于神经网络架构搜索的X射线图像违禁品检测算法

基于神经网络架构搜索的X射线图像违禁品检测算法

扫码查看
原文链接
  • 万方数据
  • 维普
为了提高卷积神经网络设计的自动化程度并进一步提高复杂背景下违禁品检测的准确率和速度,提出了一种基于神经网络架构搜索的X射线图像违禁品检测算法.首先,设计逐层渐进式搜索策略和多分支搜索空间,并基于批量归一化指标为每一个layer结构搜索最佳侧分支;然后,逐层搜索构建新的骨干网络组件;最后,组成由数据驱动的新目标检测模型.该算法在数据集HiXray、OPIXray、PIDray上分别取得了83.4%、87.2%、70.4%的检测精度.实验结果表明,本文算法能够自适应数据集并自动搜索出性能更好的Backbone组件,与FCOS、YOLOv4 等主流算法相比,有效提高了复杂背景下违禁品检测的准确率和速度.
Prohibited Item Detection Algorithm with Neural Network Architecture Search Using X-ray Images
In order to improve the automation of convolutional neural network(CNN)design and further improve the accuracy and speed of prohibited item detection in complex background,a prohibited item detection algorithm for X-ray images was proposed based on neural network architecture search.First,a layer-by-layer progressive search strategy and a multi branch search space were designed,and the best side branches were searched for each layer structure based on batch normalization(BN)metric.Then,the backbone compo-nent of the target detection model was constructed based on the layer-by-layer progressive search strategy.Finally,a new data-driven X-ray image prohibited item detection model was formed.The experimental results have demonstrated that the algorithm achieves detection accuracy of 83.4%,87.2% and 70.4%respectively on three datasets HiXray,OPIXray and PIDray.The algorithm proposed in this pa-per can adapt the dataset and automatically search for Backbone components with better performance from the dataset,and effectively im-proves the accuracy and speed of prohibited item detection compared with mainstream algorithms such as FCOS and YOLOv4.

neural network architecture searchsearch strategytarget detectionprohibited item detectionX-ray security image

成浪、敬超、陈文鹏

展开 >

桂林理工大学信息科学与工程学院,桂林 541004

桂林理工大学嵌入式技术与智能系统重点实验室,桂林 541004

桂林电子科技大学可信软件重点实验室, 桂林 541004

神经网络架构搜索 搜索策略 目标检测 违禁品检测 X射线图像

国家自然科学基金国家自然科学基金广西自然科学基金广西可信软件重点实验室基金广西中青年教师基础能力提升项目

61802085618620192020GXNSFAA159038kx2020112022KY0252

2024

10.12404/j.issn.1671-1815.2300580

科学技术与工程
中国技术经济学会

科学技术与工程

CSTPCD北大核心
影响因子：0.338
ISSN：1671-1815
年,卷(期)：2024.24(2)
  • 7