改進(jìn)YOLOv5的輕量化交通標(biāo)志檢測(cè)算法
數(shù)據(jù)采集與處理
頁(yè)數(shù): 11 2023-11-15
摘要: 隨著當(dāng)今時(shí)代科技和人工智能的高速發(fā)展,人們?cè)絹?lái)越傾向于無(wú)人駕駛這項(xiàng)技術(shù)。考慮到安全問(wèn)題,針對(duì)駕駛過(guò)程中交通標(biāo)志的實(shí)時(shí)檢測(cè)問(wèn)題,在YOLOv5模型的基礎(chǔ)上做出改進(jìn),提出了一種輕量化的交通標(biāo)志檢測(cè)算法。在模型的特征融合部分加入了注意力機(jī)制,可以使模型更加突出目標(biāo)特征。在檢測(cè)層前加入一種輕量化的亞像素卷積層,在不增加計(jì)算量的基礎(chǔ)上,有效地提高檢測(cè)特征圖的分辨率。對(duì)損失函數(shù)CIoU(C...