基于YOLOv3-tiny的二轮车头盔检测
计算机应用与软件
页数: 6 2024-05-12
摘要: 针对二轮车驾乘人员头盔佩戴问题,提出一种基于YOLOv3-tiny的轻量化头盔检测模型。将原始模型主干网络进行轻量化处理,减少检测模型的参数量,在网络中添加U型特征二次融合模块,引入关于边框距离的DIoU损失函数,用于提高检测模型的特征提取能力和识别精度。在测试集上的实验表明,改进后的模型相比原YOLOv3-tiny模型表现出更高的查全率和mAP及F1指标,且在保持较小参数量的... (共6页)