基于YOLOv5s的輕量化乒乓球目標(biāo)檢測算法
計算機(jī)應(yīng)用
頁數(shù): 6 2023-06-30
摘要: 針對乒乓球目標(biāo)檢測方法易受環(huán)境、光線、速度等多種因素干擾導(dǎo)致精度和實時性不佳的問題,提出了一種基于YOLOv5s框架的輕量化乒乓球目標(biāo)檢測算法——SYOLO5(Shuffle-YOLOv5s)。首先,采用改進(jìn)的ShuffleNetV2網(wǎng)絡(luò)單元組合重構(gòu)YOLOv5s主干網(wǎng)絡(luò),提高特征提取速度;其次,在特征融合的過程中引入高效通道注意力(ECA)機(jī)制,有效提升模型的檢測性能;接著,...