目标检测
特征(语言学)
计算机科学
人工智能
频道(广播)
水准点(测量)
棱锥(几何)
对象(语法)
计算机视觉
代表(政治)
模式识别(心理学)
数学
电信
地理
哲学
语言学
几何学
大地测量学
政治
政治学
法学
作者
Yi Zhang,Zhuo Li,Chunjie Ma,Yutong Zhang,Jiafeng Li
标识
DOI:10.1007/s11220-023-00416-7
摘要
Fast and accurate prohibited object detection in X-ray images is great challenging. Based on YOLOv6 object detection framework, in this paper, Channel-Target Attention Feature Pyramid Network (CTA-FPN) is proposed for prohibited object detection in X-ray images. It includes two key components: TAAM (Target Aware Attention Module) and CAM (Channel Attention Module). TAAM is to generate the target attention map to enhance the features of prohibited object regions and suppress those of the background regions, so as to solve the problems of object occlusion and cluttered background in X-ray images. CAM is to highlight the feature channels important to the detection tasks, and suppress the irrelevant ones. The target-wise and channel-wise feature enhancement can effectively strengthen the feature representation capability of the network. The proposed CTA-FPN is incorporated into S, M and L models of YOLOv6 respectively, obtaining three X-ray prohibited object detection models. The experimental results on two publicly available benchmark datasets of SIXray and CLCXray show that, CTA-FPN can effectively improve the detection performance of YOLOv6. Especially, YOLOv6-CTA-FPN-L can achieve the state-of-the-arts detection accuracy.
科研通智能强力驱动
Strongly Powered by AbleSci AI