基于改进YOLOv5的驾驶员手持手机检测算法研究

针对驾驶员手持手机行为检测精度低问题,提出一种改进的驾驶员手持手机行为检测算法.首先,在YOLOv5骨干网络中引入改进的注意力机制模块,更好地获取上下文信息,提高小目标检测的精确度.其次,采用一种改进的特征融合方法,提取三个尺度的特征,并对特征进行融合,更好地提取局部信息.实验结果表明,与YOLOv5 相比,该检测算法在自制数据集上的精确度达到71.9％,提高了2.1％,对小目标的检测效果显著.