基于改進YOLOv5的內(nèi)河道船舶檢測方法研究
遼寧工程技術(shù)大學學報(自然科學版)
頁數(shù): 8 2024-12-15
摘要: 為解決內(nèi)河航道通航時船舶相互遮擋導致的錯檢和漏檢問題,提出一種基于改進YOLOv5模型的內(nèi)河道船舶檢測方法。該方法采用C2f模塊以捕捉和融合多尺度特征,增強低級特征的語義信息;引入Shuffle Attention模塊強化特征表示,使模型能夠聚焦于信息量更大的區(qū)域,并有效抑制無關(guān)特征;同時,采用Wise-IoU損失函數(shù),有效防止低質(zhì)量錨框產(chǎn)生有害梯度,加速模型的優(yōu)化過程。研究結(jié)... (共8頁)