Suppr超能文献

配方无人竞赛中圆锥桶检测的改进算法研究

Research on Improved Algorithms for Cone Bucket Detection in Formula Unmanned Competition.

作者信息

Li Xu, Li Gang, Zhang Zhe, Sun Haosen

机构信息

School of Automobile and Traffic Engineering, Liaoning University of Technology, Jinzhou 121001, China.

出版信息

Sensors (Basel). 2024 Sep 13;24(18):5945. doi: 10.3390/s24185945.

Abstract

The model network based on YOLOv8 for detecting race cones and buckets in the Formula Unmanned Competition for Chinese university students needs help with problems with complex structure, redundant number of parameters, and computation, significantly affecting detection efficiency. A lightweight detection model based on YOLOv8 is proposed to address these problems. The model includes improving the backbone network, neck network, and detection head, as well as introducing knowledge distillation and other techniques to construct a lightweight model. The specific improvements are as follows: firstly, the backbone network for extracting features is improved by introducing the ADown module in YOLOv9 to replace the convolution module used for downsampling in the YOLOv8 network, and secondly, the FasterBlock in FasterNet network was introduced to replace the fusion module in YOLOv8 C2f, and then the self-developed lightweight detection head was introduced to improve the detection performance while achieving lightweight. Finally, the detection performance was further improved by knowledge distillation. The experimental results on the public dataset FSACOCO show that the improved model's accuracy, recall, and average precision are 92.7%, 84.6%, and 91%, respectively. Compared with the original YOLOv8n detection model, the recall and average precision increase by 2.7 and 1.2 percentage points, the memory is half the original, and the model computation is 51%. The model significantly reduces the misdetection and leakage of conical buckets in real-vehicle tests and, at the same time, ensures the detection speed to satisfy the deployment requirements on tiny devices. Satisfies all the requirements for deployment of tiny devices in the race car of the China University Student Driverless Formula Competition. The improved method in this paper can be applied to conebucket detection in complex scenarios, and the improved idea can be carried over to the detection of other small targets.

摘要

针对中国大学生方程式无人驾驶竞赛中用于检测赛道锥桶和桶的基于YOLOv8的模型网络,存在结构复杂、参数冗余和计算量大的问题,严重影响检测效率。为此提出了一种基于YOLOv8的轻量级检测模型来解决这些问题。该模型包括改进骨干网络、颈部网络和检测头,以及引入知识蒸馏等技术来构建轻量级模型。具体改进如下:首先,通过引入YOLOv9中的ADown模块来改进用于提取特征的骨干网络,以取代YOLOv8网络中用于下采样的卷积模块;其次,引入FasterNet网络中的FasterBlock来取代YOLOv8 C2f中的融合模块,然后引入自主研发的轻量级检测头,在实现轻量级的同时提高检测性能。最后,通过知识蒸馏进一步提高检测性能。在公共数据集FSACOCO上的实验结果表明,改进后的模型准确率、召回率和平均精度分别为92.7%、84.6%和91%。与原始的YOLOv8n检测模型相比,召回率和平均精度分别提高了2.7和1.2个百分点,内存为原来的一半,模型计算量为51%。该模型在实车测试中显著减少了锥桶的误检和漏检,同时确保了检测速度,满足在微小设备上的部署要求。满足中国大学生无人驾驶方程式赛车中微小设备部署的所有要求。本文提出的改进方法可应用于复杂场景下的锥桶检测,且改进思路可推广到其他小目标的检测。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/59ed/11435763/b4953e8dec46/sensors-24-05945-g001.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验