基于改進(jìn)YOLOv5的軸類零件表面缺陷檢測算法

摘要： 針對當(dāng)前軸件表面缺陷種類繁多、缺陷形態(tài)復(fù)雜等原因?qū)е碌臋z測精度低，提出了一種改進(jìn)的YOLOv5的目標(biāo)軸件表面缺陷檢測方法。為解決在日常生產(chǎn)中經(jīng)常出現(xiàn)軸件表面小目標(biāo)缺陷被漏檢、錯(cuò)檢的問題，在原YOLOv5基礎(chǔ)上，添加一個(gè)新的小目標(biāo)檢測層，并將較淺特征圖與深特征圖拼接，使得整個(gè)網(wǎng)絡(luò)更加關(guān)注小目標(biāo)缺陷。同時(shí)為解決多目標(biāo)缺陷和不完整軸件檢測精度低與漏檢，自建數(shù)據(jù)集中添加多目標(biāo)缺陷與遮擋...