基于改進(jìn)YOLOv5的高鐵周界入侵人員檢測(cè)方法研究
鐵道標(biāo)準(zhǔn)設(shè)計(jì)
頁數(shù): 8 2022-07-19
摘要: 針對(duì)高鐵周界入侵人員檢測(cè)研究中,存在遠(yuǎn)距離入侵目標(biāo)識(shí)別效果差的問題,提出一種基于改進(jìn)YOLOv5的高鐵周界入侵人員檢測(cè)方法。首先,在YOLOv5s模型的Neck網(wǎng)絡(luò)層融合坐標(biāo)注意力機(jī)制,以提升模型的特征提取能力;其次,增加一個(gè)預(yù)測(cè)層來提升對(duì)遠(yuǎn)距離入侵人員的檢測(cè)性能;進(jìn)一步地,利用K-means聚類算法得到數(shù)據(jù)集合適的anchor框;最后,改進(jìn)邊界框回歸損失函數(shù)以提高邊界框的定位...