考虑样本不平衡的X光安检图像违禁品分类方法 |
| |
引用本文: | 冯霞,魏新坤,刘才华,赫鑫宇.考虑样本不平衡的X光安检图像违禁品分类方法[J].北京航空航天大学学报,2023(12):3215-3221. |
| |
作者姓名: | 冯霞 魏新坤 刘才华 赫鑫宇 |
| |
作者单位: | 1. 中国民航大学计算机科学与技术学院;2. 民航智慧机场理论与系统重点实验室 |
| |
基金项目: | 天津市教委科研计划(2021KJ037);;中央高校基本科研业务费专项资金(3122021052)~~; |
| |
摘 要: | X光安检图像违禁品分类被广泛应用于协助维护航空和运输安全。针对X光安检图像中违禁品尺度不一、存在困难样本及旅客行李安检固有的正负样本不均衡等问题,提出一种端到端的考虑样本不平衡的X光安检图像违禁品分类方法。采用多尺度特征提取网络捕获尺度不一的多类型违禁品特征,通过特征融合模块提升模型对图像边缘和纹理特征的表达能力,基于代价敏感思想设计损失函数,解决数据集不平衡问题,并提高困难样本分类精准度。在公开数据集SIXray上构建的子集实验结果表明:所提方法相较于端到端分类模型,平均AP指标值提升了4.5%,特别是对剪刀等难分类样本,AP指标值都有显著的提升效果。
|
关 键 词: | 违禁品分类 样本不平衡 X光图像 多尺度 困难样本分类 代价敏感 |
|