基于改進(jìn)YOLOv4-Tiny的河湖船舶目標(biāo)檢測(cè)算法
人民長(zhǎng)江
頁(yè)數(shù): 8 2023-09-25
摘要: 為有效提升復(fù)雜河湖環(huán)境下河湖過(guò)往船舶的精準(zhǔn)快速識(shí)別效果,提出了基于改進(jìn)YOLOv4-Tiny的河湖船舶目標(biāo)檢測(cè)算法。該算法首先通過(guò)引入Sigmoid加權(quán)線性單元(SiLU)激活函數(shù),構(gòu)建卷積+批標(biāo)準(zhǔn)化+SiLU激活函數(shù)的卷積模塊,并替換主干網(wǎng)絡(luò)中原有模塊,然后在主干網(wǎng)絡(luò)之后增加改進(jìn)的空間金字塔池化(SPP)網(wǎng)絡(luò),最后在特征金字塔網(wǎng)絡(luò)(FPN)中引入卷積塊注意力模塊(CBAM),...