一种增强的违禁物品识别模型
计算机视觉与模式识别
2021-08-11 v2
摘要
我们提出了一种新的建模方法，以通过X射线图像提升违禁物品识别的性能。我们分析了违禁物品与X射线图像的特点，发现部分物品尺度过小难以被识别，制约了模型性能。随后我们采用一组的数据增强并改进模型以适应违禁物品识别领域。模型中集成了卷积块注意力模块（CBAM）与重打分机制。经改进，我们的模型在SIXray10上取得了89.9%的mAP，在SIXray10上取得了74.8%的mAP。
引用
@article{arxiv.2102.12256,
title = {An Enhanced Prohibited Items Recognition Model},
author = {Tianze Rong and Hongxiang Cai and Yichao Xiong},
journal= {arXiv preprint arXiv:2102.12256},
year = {2021}
}