海洋環(huán)境下基于增強YOLOv7的垃圾目標檢測
所屬分類:技術(shù)論文
上傳者:wwei
文檔大?。?span>3800 K
標簽: EUIE eSE注意力 海洋垃圾檢測
所需積分:0分積分不夠怎么辦?
文檔介紹:針對海洋垃圾識別任務(wù)在實際應(yīng)用中模型準確率不高的問題,提出一種基于優(yōu)化YOLOv7的海洋垃圾識別算法。在圖像增強部分,基于概率UIE的框架,通過添加eSE注意力減少特征信息的丟失。在損失函數(shù)部分,在IoU損失函數(shù)的基礎(chǔ)上引入兩層注意力機制的損失函數(shù),將其與EIoU損失函數(shù)融合進一步提升模型的泛化能力。將該算法應(yīng)用于海洋垃圾檢測任務(wù),并在基礎(chǔ)數(shù)據(jù)集上對其進行評估。在YOLOTrashCan兩個數(shù)據(jù)集上的平均精度均值指標分別達到69.5%、63.5%,相較于YOLOv7算法分別提升6%、1.6%。整體實驗結(jié)果表明,所構(gòu)建的算法能有效提升海洋垃圾檢測的準確性。
現(xiàn)在下載
VIP會員,AET專家下載不扣分;重復(fù)下載不扣分,本人上傳資源不扣分。