公开(公告)号：CN118314514A

公开(公告)日：2024-07-09

申请号：CN202410468644.7

申请日：2024-04-18

Applicant: 江苏大学

Inventor： 宋雪桦 ,  郑志林 ,  王玉周 ,  金华 ,  王昌达 ,  袁昕

IPC: G06V20/52 ,  G06V10/774 ,  G06V10/776 ,  G06V10/80 ,  G06V10/82 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明涉及一种面向智慧安检的违禁品检测方法，在主干网络添加包含无参数的SimAM模块的SELAN，使模型在特征提取时能同时关注空间和通道上的信息；在颈部网络添加包含BRA模块的PBELAN，利用稀疏注意力机制在计算量较小的情况下，使模型能聚焦于特征图中最相关的像素点，使用部分卷积来降低模型参数量；使用WIOU函数作为边界框损失函数，解决质量较好和较差的数据集样本之间的平衡问题，提高模型泛化能力。本发明采用公开数据集CLCray作为数据集进行训练，并将训练好的模型权重用于检测，有效降低了安检对人力和财力的投入，在检测率提高的同时降低了模型参数量，进而较好地满足公共场合安检违禁品检测任务的需求。