當前位置:首頁 > 科技文檔 > 鐵路運輸 > 正文

改進YOLOV5s的鐵軌裂紋目標檢測算法

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

開通會員,享受整站包年服務(wù)立即開通 >