基于YOLOv5的紅外船舶目標(biāo)檢測(cè)算法
紅外與激光工程
頁(yè)數(shù): 12 2023-10-25
摘要: 針對(duì)紅外船舶目標(biāo)在海上復(fù)雜海天背景下檢測(cè)困難,且數(shù)據(jù)集目標(biāo)大小與錨框不符造成的算法邊界回歸效果差、檢測(cè)不準(zhǔn)確等問(wèn)題,提出了一種基于改進(jìn)YOLOv5的紅外船舶目標(biāo)檢測(cè)算法。首先針對(duì)錨框與數(shù)據(jù)集目標(biāo)形狀不匹配問(wèn)題,通過(guò)改變K-means++聚類(lèi)算法選取簇中心的評(píng)價(jià)標(biāo)準(zhǔn),使用中位數(shù)代替平均數(shù)來(lái)決定簇中心,改進(jìn)了錨框算法,使得錨框與船舶目標(biāo)更加匹配,提高了算法的平均檢測(cè)精度。改進(jìn)后的聚...