摘要
本发明公开了一种基于特征图重赋权的X光违禁品包裹检测方法和装置,通过构建X光违禁品训练集,依次获取图像特征图、感兴趣区域特征图、类注意特征图,采用类注意特征图对感兴趣区域特征图做重赋权操作,得到单维特征向量,确定各类违禁品的预测类别,调节下一次训练过程中采用的特征提取网络、元特征提取器及分类回归网络,返回执行依次获取图像特征图、感兴趣区域特征图、类注意特征图的过程,重新将X光违禁品训练集输入调节后的特征提取网络进行迭代训练,直至训练次数大于或等于预设的迭代次数,确定违禁品检测器,将待检测包裹输入违禁品检测器,以获取违禁品检测器输出的违禁品位置及违禁品类别,达到节省人力物力,降低检测成本的目的。
- 单位