RFSD-YOLO: An Enhanced X-Ray Object Detection Model for Prohibited Items
X-ray image detection is essential for ensuring public safety, but traditional methods rely heavily on human analysis and are relatively inefficient. To address this issue, this paper proposes a new object detection model called RFSD-YOLO, which is based on the YOLOv8 model and is specifically designed for detecting prohibited items in X-ray images. The model adopts the RFCAConv structure instead of the conventional convolutional operation. This allows for independent parameterisation among the convolutional kernels, enhancing the model's ability to capture and express potential prohibited item features. The neck section of the model includes the GSConv and VoVgscspdesigns, which aim to balance complexity and parameter size while maintaining detection performance and reducing computational burden. Additionally, we have introduced a dynamic head structure, DyHead, which replaces the traditional detection head design and improves detection accuracy without adding computational cost. The experimental results demonstrate that our enhanced model surpasses the current state-of-the-art object detection models in detecting prohibited items. Additionally, we introduce a simplified version of the RFSD-YOLOnano model to cater to resource-constrained environments. This streamlined model improves AP, AR, and mAP by 4.4%, 14.6%, and 3.4%, respectively, compared to YOLOv8n. This series of innovations not only validates the superiority of our model, but also provides new solutions in the field of automated detection of prohibited items for X-ray security screening.
Authors
- Yuechen Zhang (ORCID: https://orcid.org/0009-0000-9112-0216)
- Aimin Li (ORCID: https://orcid.org/0000-0003-3687-4378)
- Zhiyao Li (ORCID: https://orcid.org/0009-0004-0137-6760)
- Xiaotong Kong (ORCID: https://orcid.org/0000-0002-3540-6233)
- Wenqiang Li (ORCID: https://orcid.org/0000-0003-3696-1893)
Institutions
- Qilu University of Technology (CN)
- Shandong Academy of Sciences (CN)
Publication Details
- Published
- 2024-10-06
- DOI
- https://doi.org/10.1109/smc54092.2024.10831942
- Primary Topic
- Radiomics and Machine Learning in Medical Imaging
- Type
- article
- Field-Weighted Citation Impact
- 0.00