基于輕量化模型的鋼軌扣件缺陷檢測(cè)系統(tǒng)

摘要： 鐵路巡檢工作中，嵌入式設(shè)備受算力和存儲(chǔ)空間的限制，存在使用YOLO V5模型檢測(cè)鋼軌扣件缺陷速度慢、精度較低的問(wèn)題。通過(guò)替換YOLO V5主干卷積網(wǎng)絡(luò)為MobileNet V3,將網(wǎng)絡(luò)中的激活函數(shù)修改為Mish并融合協(xié)同注意力機(jī)制，實(shí)現(xiàn)模型的輕量化改進(jìn)。將改進(jìn)后的模型部署到嵌入式設(shè)備Jetson TX2上，使用板載CSI攝像頭掃描、拍攝鋼軌扣件，并搭載顯示屏等設(shè)備構(gòu)成鋼軌扣件缺...