改進(jìn)YOLOV5s的鐵軌裂紋目標(biāo)檢測算法
計(jì)算機(jī)工程與應(yīng)用
頁數(shù): 9 2023-08-25
摘要: 鐵軌表面的裂紋圖像目標(biāo)小、背景干擾信息多、分辨率高,使其目標(biāo)檢測效果并不理想。對此,提出一種改進(jìn)YOLOV5s的目標(biāo)檢測算法以提升鐵軌裂紋的檢測效果。在頸部網(wǎng)絡(luò)中引入VOV-GSCSP模塊,用更輕量的卷積方式GSconv來替換普通的卷積,在保留更多細(xì)節(jié)信息的同時(shí)減輕網(wǎng)絡(luò)的計(jì)算量。改進(jìn)特征金字塔結(jié)構(gòu),提出多路徑跨層融合結(jié)構(gòu),在特征金字塔下采樣的過程中跨層融入主干網(wǎng)絡(luò)的信息,保留更... (共9頁)