基于KRB-YOLOv5s的煤矸識別方法
遼寧工程技術大學學報(自然科學版)
頁數(shù): 8 2024-08-15
摘要: 為解決煤礦高粉塵、低照度、高噪聲與堆疊等復雜環(huán)境因素導致的煤矸識別精度低、漏檢與誤檢問題,提出一種基于KRB-YOLOv5s算法的煤矸識別方法。采用K均值聚類(K-means++)算法對數(shù)據(jù)集進行重新聚類,以得到更精準的錨框參數(shù);在YOLOv5s主干網(wǎng)絡中引入大核卷積結構重參數(shù)(RepLKNet)網(wǎng)絡,通過大核卷積架構提取目標更高層級的特征信息;在YOLOv5s頸部引入加權雙向... (共8頁)