摘要
为了提高交通标志检测的实时性和可行性,提出了一种基于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%。该模型具有检测精度高、参数量少、计算量低、速度快等优点,较原算法有很大地提升,且与其他先进的交通标志检测模型比较时表现出了很强的竞争力,在交通标志检测中具有较大优势。
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.
作者
朱强军
胡斌
汪慧兰
王杨
ZHU Qiangjun;HU Bin;WANG Huilan;WANG Yang(Department of Big Data and Artificial Intelligence,Wanjiang College of Anhui Normal University,Wuhu Anhui 241000,China;School of Physics and Electronic Information,Anhui Normal University,Wuhu Anhui 241000,China;School of Computer and Information,Anhui Normal University,Wuhu Anhui 241000,China)
出处
《图学学报》
CSCD
北大核心
2024年第3期422-432,共11页
Journal of Graphics
基金
安徽省高校自然科学研究重点项目(2023AH052459)
安徽师范大学皖江学院重点自然科研项目(WJKYZD-202301)
安徽省高等学校省级质量工程项目(2022sx052)
安徽师范大学皖江学院教学质量工程项目(WJXGK-202201)。