基于轻量化YOLOv8s交通标志的检测
Detection of traffic signs based on lightweight YOLOv8s
朱强军 1胡斌 1汪慧兰 2王杨3
作者信息
- 1. 安徽师范大学皖江学院大数据与人工智能系,安徽 芜湖 241000
- 2. 安徽师范大学物理与电子信息学院,安徽 芜湖 241000
- 3. 安徽师范大学计算机与信息学院,安徽 芜湖 241000
- 折叠
摘要
为了提高交通标志检测的实时性和可行性,提出了一种基于 YOLOv8s 的轻量化交通标志检测模型.首先,用FasterNet中的残差模块FasterNetBlock替换C2f模块中的BottleNeck,降低模型参数量和计算量;其次,用一种小目标检测层去替换大目标检测层,降低 Backbone 中网络层数,实现大幅度提高检测速度和降低参数量;最后,用Wise-IOU替换原CIOU损失函数,提高速度和精度.在TT100K交通标志数据集上验证,其与YOLOv8s模型比较,mAP50提高了5.16%,参数量降低了76.48%,计算量降低了13.33%,FPS快了 35.83%.与其他模型相比,mAP50 平均提高了 15.11%,参数量平均降低了 85.74%,计算量平均下降了 46.23%,FPS 平均提高了 31.49%.该模型具有检测精度高、参数量少、计算量低、速度快等优点,较原算法有很大地提升,且与其他先进的交通标志检测模型比较时表现出了很强的竞争力,在交通标志检测中具有较大优势.
Abstract
To enhance the real-time capability and feasibility of traffic sign detection,a lightweight traffic sign detection model based on YOLOv8s was proposed.Firstly,the BottleNeck in the C2f module was replaced with the residual module FasterNetBlock in FasterNet,reducing the model's parameter count and computational complexity.Secondly,the large object detection layer was replaced with a small object detection layer,decreasing the number of network layers in Backbone and achieving a significant improvement in detection speed and a reduction in parameter count.Finally,the original complete intersection over union(CIOU)loss function was replaced with the wise intersection over union(Wise-IOU),thereby enhancing both speed and accuracy.Verified on the TT100K traffic sign dataset,compared with the YOLOv8s model,mAP50 increased by 5.16%,parameter count decreased by 76.48%,computational complexity decreased by 13.33%,and frames per second(FPS)improved by 35.83%.In comparison to other models,mAP50 exhibited an average increase of 15.11%,an average decrease of 85.74%in parameter count,an average decrease of 46.23%in computational complexity,and an average increase of 31.49%in FPS.This model achieved the advantages of high detection accuracy,small number of parameters,low computational complexity,and fast speed.It represented a substantial improvement over the original algorithm and demonstrated strong competitiveness when compared to other advanced traffic sign detection models,with great advantages in traffic sign detection.
关键词
轻量化/YOLOv8s/改进小目标层/交通标志检测/Wise-IOU/TT100KKey words
lightweight/YOLOv8s/improved small target layer/traffic sign detection/Wise-IOU/TT100K引用本文复制引用
基金项目
安徽省高等学校自然科学研究重点项目(2023AH052459)
安徽师范大学皖江学院重点自然科研项目(WJKYZD-202301)
安徽省高等学校省级质量工程项目(2022sx052)
安徽师范大学皖江学院教学质量工程项目(WJXGK-202201)
出版年
2024
NETL
NSTL
万方数据