摘要
本发明公开了基于生成式稀疏性的X光安检图像违禁品检测方法,该方法首先获取含有目标边界框和类别标注的X光安检图像数据集,先后利用ResNet和特征金字塔结构提取安检图像中的多级特征。其次利用扩散模型生成带噪声的预案框,利用预案框生成感兴趣区域特征以及具有上下文感知的特征表示;通过动态过滤融合模块,生成精炼特征图,得到目标预测框和对应类别分数。然后在标签分配阶段,利用匈牙利Top‑K匹配策略,通过全局性匹配,为每个真值寻找K个同等重要的最优正样本,并构建损失函数,进行迭代训练,并在验证集上验证,输出违禁品检测效果图。本发明无论是在X光违禁品检测精准度,还是在时效性都表现良好。
技术关键词
X光安检图像
违禁品检测方法
多级特征
生成感兴趣区域
特征金字塔
交叉注意力机制
样本
匈牙利算法
噪声
生成特征
阶段
动态
参数
标签
数据
时效性