基于YOLOv3-tiny的二輪車頭盔檢測

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