| 82 | 5 | 0 |
阅读 |
下载 |
被引 |
针对现有X光图像违禁品检测精度低、速度慢且公共数据集样本过少等问题,提出一种基于CycleGAN和改进YOLOv8n的X光图像违禁品检测算法。该算法利用CycleGAN模型生成更多具有安检X光风格特点的违禁品图像,建立足够多的违禁品图像样本;同时,针对X 光违禁品图像目标大小不同的特点,设计了混合卷积注意力(HCA)机制,有效加强YOLOv8n 对目标关键特征的提取能力。对于安检X 光违禁品图像特征信息不足的问题,利用CARAFE 模块替代原YOLOv8n的上采样方式,并采用BiFPN对颈部网络进行改进,通过有效的特征融合方法和上采样方式弥补图像特征信息不足的问题。最后针对YOLOv8n检测头参数冗余问题,使用参数共享和DBB模块减少模型参数量,同时保证模型的检测精度。在经过扩充的SIXary数据集上对所提出的检测算法进行测试,实验结果表明,该算法的mAP值达到93.1%,相比原始算法提升2.9%,同时参数量减少8.5%,并且保持了与原始模型相当的检测速度。该算法的性能相比原始算法取得显著提升,验证了其改进的有效性。
Abstract:In allusion to the problems of low accuracy, slow detection speed and too few samples in public data set, an X⁃ray image contraband detection algorithm based on CycleGAN and improved YOLOv8n is proposed. In this model, the CycleGAN model is used to generate more contraband images with security X⁃ray style characteristics, and establish enough samples of contraband image. According to the characteristics of different sizes of objects in X ⁃ ray contraband images, the hybrid convolutional attention (HCA) mechanism is designed to effectively enhance the ability of YOLOv8n to extract key features of objects. In allusion to the problem of insufficient feature information of security X⁃ray contraband images, CARAFE module is used to replace the original YOLOv8n upsampling mode, BiFPN network is used to improve the neck network, and the effective feature fusion method and upsampling method are used to made up the insufficient image feature information. Aiming at the problem of parameter redundancy of YOLOv8n detection head, parameter sharing and DBB module are used to reduce the number of model parameters and ensure the detection accuracy of the model. The detection algorithm was tested on the expanded SIXary dataset. The experimental results show that the mAP value of this algorithm can reach 93.1, which is 2.9% higher than that of the original model, while the number of parameters is decreased by 8.5%, maintaining the detection speed of original model. The performance of this algorithm is significant improved compared to the original algorithm, verifying the effectiveness of its improvement.
[1] 孟蕊.安检数智化+网络安全防护航空货物运输高效安全的标配[J].中国安防,2024(3):70⁃72.
[2] WEI Y, WANG H. Cooperative distillation with X⁃ray images classifiers for prohibited items detection [J]. Engineering applications of artificial intelligence, 2024, 127: 107276.
[3] REN Y, ZHANG H, SUN H, et al. LightRay: lightweight network for prohibited items detection in X⁃ray images during security inspection [J]. Computers and electrical engineering, 2022, 103: 108283.
[4] XIANG N, GONG Z, XU Y, et al. Material ⁃ aware path aggregation network and shape decoupled SIoU for X ⁃ ray contraband detection [J]. Electronics, 2023, 12(5): 1179.
[5] WANG B, DING H, CHEN C. AC⁃YOLOv4: an object detection model incorporating attention mechanism and atrous convolution for contraband detection in x ⁃ ray images [J]. Multimedia tools and applications, 2023, 83(9): 1⁃20.
[6] ZHU J Y, PARK T, ISOLA P, et al. Unpaired image⁃to⁃image translation using cycle ⁃ consistent adversarial networks [C]// Proceedings of the IEEE international conference on computer vision. [S.l.]: IEEE, 2017: 2223⁃2232.
[7] 李文强,陈莉,谢旭,等.改进YOLOv5的X光图像违禁品检测算法[J].计算机工程与应用,2023,59(16):170⁃176.
[8] 袁金豪,张南峰,阮洁珊,等.基于改进YOLOX 算法的X 射线图像违禁品检测方法[J].激光技术,2023,47(4):547⁃552.
[9] 康佳楠,张良.多通道区域建议的多尺度X 光安检图像检测[J].计算机工程与应用,2022,58(1):224⁃231.
[10] 赵彤欣.基于风格迁移的安检X光训练图像仿真与评价方法研究[D].天津:中国民航大学,2022.
[11] PATHAK D, RAJU U S N. Content ⁃ based image retrieval using feature⁃fusion of Group normalized⁃inception⁃darknet⁃53 features and handcraft features [J]. Optik, 2021, 246: 167754. [12] ZENG W, HE M. Rice disease segmentation method based on CBAM⁃CARAFE⁃DeepLabv3+ [J]. Crop protection, 2024, 180: 106665.
[13] DING X, ZHANG X, HAN J, et al. Diverse branch block: building a convolution as an inception ⁃ like unit [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Vancouver, BC, Canada: IEEE, 2021: 10886⁃10895.
[14] 郝帅,赵新生,马旭,等.基于TR⁃YOLOv5的输电线路多类缺陷目标检测方法[J].图学学报,2023,44(4):667⁃676.
[15] 谷雨,赵军.列车闸瓦钎及闸瓦故障图像检测算法研究[J].图学学报,2023,44(1):88⁃94.
[16] FENG C, ZHONG Y, GAO Y, et al. Tood: task⁃aligned one⁃ stage object detection [C]// 2021 IEEE/CVF International Conference on Computer Vision (ICCV). [S. l.]: IEEE, 2021: 3490⁃3499.
[17] TIAN Z, CHU X, WANG X, et al. Fully convolutional one⁃ stage 3D object detection on lidar range images [J]. Advances in neural information processing systems, 2022, 35: 34899 ⁃ 34911.
基本信息:
DOI:10.16652/j.issn.1004⁃373x.2026.06.015
引用信息:
[1]左嘉炜,魏培旭,葛超.基于CycleGAN和改进YOLOv8n的X光违禁品检测[J],2026,49(6):94⁃101.DOI:10.16652/j.issn.1004⁃373x.2026.06.015.
基金信息:
河北省自然科学基金项目(F2021209006)
阅读
下载
被引