YOLOv7-F:一種輕量級(jí)船舶實(shí)時(shí)檢測(cè)算法
兵器裝備工程學(xué)報(bào)
頁(yè)數(shù): 8 2024-11-15
摘要: 船舶檢測(cè)在內(nèi)河航運(yùn)管理中至關(guān)重要,在復(fù)雜的水面條件下,船舶檢測(cè)很難兼顧準(zhǔn)確性和實(shí)時(shí)性。針對(duì)這個(gè)問(wèn)題,提出了一種改進(jìn)YOLOv7的船舶實(shí)時(shí)檢測(cè)方法YOLOv7-F。YOLOv7-F將GhostNet引入骨干網(wǎng)絡(luò)進(jìn)行特征提取,再將分布移位卷積引入特征融合網(wǎng)絡(luò),實(shí)現(xiàn)了模型輕量化。在特征融合網(wǎng)絡(luò)中引入注意力機(jī)制,彌補(bǔ)模型輕量化帶來(lái)的精度損失。損失函數(shù)也進(jìn)行相應(yīng)改進(jìn),使檢測(cè)模型更適用于船... (共8頁(yè))