DS-YOLOv4-tiny救援机器人目标检测模型

• 作者: 杨艳红 钟宝江 田宏伟
• 来源期刊: 计算机仿真
• 学科: 工学
• 关键词: 救援机器人 目标检测 仿真
• 年，卷（期）: 2022,（1）
• 所属期刊栏目: 仿真智能化
• 页码范围: 387-393
• 页数: 7页
• 分类号: TP183|TP391.41
• 语种: 中文
• DOI: 10.3969/j.issn.1006-9348.2022.01.081

摘要

针对救援机器人对救援目标检测实时性和准确度要求比较高的问题,提出一种改进的YOLOv4-tiny救援机器人目标检测模型DS-YOLOv4-tiny.改进YOLOv4-tiny的网络结构,将骨干网络中的CSP改为DenseNet模块以增强特征复用,同时大量减少了网络参数量,使训练后的权重文件更小;针对YOLOv4-tiny深层网络无法准确提取有效特征信息的问题,在颈部网络中增加SPP模块以得到不同感受野的特征,提高定位准确性;通过K-means聚类算法获取锚框以适应受灾人员检测,提高算法精度.将训练后的模型部署在嵌入式NVIDIA Jetson Nano平台上进行测试,实验结果表明:DS-YOLOv4-tiny检测速度快、准确率高.与YOLOv4-tiny比较,AP值达到93.47％提高了3.99％,FPS达到15.97提高了5.7％,权重文件大小降低13.39％,满足救援机器人目标检测的实时性和准确度要求.