光通信技术2024,Vol.48Issue(3):18-22,5.DOI:10.13921/j.cnki.issn1002-5561.2024.03.004
一种用于可见光通信信号调制格式识别的改进YOLOv5s算法
Improved YOLOv5s algorithm for modulation format recognition of visible light communication signal
摘要
Abstract
Aiming at the problem that modulation format recognition accuracy is susceptible to factors such as channel environ-ment and background noise interference in visible light communication signal transmission,this paper proposes an improved YOLOv5s(You Only Look Once)algorithm for modulation format recognition of visible light communication signals.Firstly,the Mixup data augmentation method is introduced at the input end of the YOLOv5s algorithm network,and it is combined with the Mosaic data augmentation method in the original network to enhance the robustness of the network and improve the generaliza-tion ability of the algorithm among different modulation format signals.Secondly,adaptively spatial feature fusion(ASFF)is in-troduced into the Neck network to fully extract features from different levels and improve detection accuracy.The experimental results indicate that under mixed signal-to-noise ratio conditions,the mean average precision(mAP)of the proposed improved al-gorithm reaches 0.903,representing a 0.7%improvement compared to the original YOLOv5s algorithm.Furthermore,the mAP reaches a high of 0.993 when the signal-to-noise ratio is 20 dB.关键词
可见光通信/调制格式识别/YOLOv5s/Mixup数据增强/自适应空间特征融合Key words
visible light communication/modulation format recognition/you only look once/Mixup data augmentation/adap-tively spatial feature fusion分类
信息技术与安全科学引用本文复制引用
王业恒,吴彰,赵永胜,严志远,毛瑞霞,朱宏娜..一种用于可见光通信信号调制格式识别的改进YOLOv5s算法[J].光通信技术,2024,48(3):18-22,5.基金项目
中央高校基本科研业务费专项资金项目(202310613083)资助. (202310613083)