基于YOLOv3-tiny的二輪車頭盔檢測(cè)
計(jì)算機(jī)應(yīng)用與軟件
頁數(shù): 6 2024-05-12
摘要: 針對(duì)二輪車駕乘人員頭盔佩戴問題,提出一種基于YOLOv3-tiny的輕量化頭盔檢測(cè)模型。將原始模型主干網(wǎng)絡(luò)進(jìn)行輕量化處理,減少檢測(cè)模型的參數(shù)量,在網(wǎng)絡(luò)中添加U型特征二次融合模塊,引入關(guān)于邊框距離的DIoU損失函數(shù),用于提高檢測(cè)模型的特征提取能力和識(shí)別精度。在測(cè)試集上的實(shí)驗(yàn)表明,改進(jìn)后的模型相比原YOLOv3-tiny模型表現(xiàn)出更高的查全率和mAP及F1指標(biāo),且在保持較小參數(shù)量的... (共6頁)