基于VSD-YOLOv5s的輕量化注塑齒輪缺陷檢測(cè)
組合機(jī)床與自動(dòng)化加工技術(shù)
頁(yè)數(shù): 5 2024-04-19
摘要: 針對(duì)注塑齒輪加工過(guò)程中在線缺陷檢測(cè)存在的識(shí)別速度較慢、識(shí)別精度較低等問(wèn)題,提出了一種基于ShuffleNetV2主干網(wǎng)絡(luò)的改進(jìn)YOLOv5s網(wǎng)絡(luò)模型。在YOLOv5s網(wǎng)絡(luò)模型的基礎(chǔ)上將原來(lái)的C3主干網(wǎng)絡(luò)結(jié)構(gòu)替換為輕量化ShuffleNetV2結(jié)構(gòu),降低模型參數(shù)量和尺寸,提升識(shí)別速度。引入SE注意力機(jī)制,并通過(guò)DIOU-NMS方法去除冗余框,減少錯(cuò)誤抑制,提高識(shí)別精度。實(shí)驗(yàn)結(jié)果表... (共5頁(yè))