关键词:
目标检测
电动车识别
YOLOv8
摘要:
随着城市化步伐的不断加快,电动车已成为城市居民日常出行的重要工具,许多居民为了充电方便,选择将电动车搬进居民楼,并通过电梯运输,这一行为导致了居民楼内电动车安全事故频发。因此,开发一种高效且准确的电梯内电动车检测算法显得尤为重要。首先,使用FasterNet中的FasterNet Block替换C2f中的Bottleneck,其降低了模型的计算量,提升了检测速度。在YOLOv8原结构中引入SEAttention注意力机制,把重要的特征进行强化来提升准确率。替换损失函数为Inner-CIoU损失函数,以提升模型检测性能和泛化能力。经过实验验证,改进后的YOLO-FSI模型在电梯内电动车数据集上的mAP50为93.7%,相较于原模型,参数量减少了23.3%,检测速度更是提升了5.2帧/秒。综上所述,YOLO-FSI模型可以有效提升电梯内电动车的检测能力,并且做到了轻量化及快速推理。