基于边缘提取与空洞卷积的抓取目标检测算法 |
| |
引用本文: | 张晓寒,张文彬,赵景波.基于边缘提取与空洞卷积的抓取目标检测算法[J].控制工程,2024(2):359-365. |
| |
作者姓名: | 张晓寒 张文彬 赵景波 |
| |
作者单位: | 青岛理工大学信息与控制工程学院 |
| |
基金项目: | 国家自然科学基金资助项目(51475251);;山东省自然科学基金资助项目(ZR2013FM014);;山东省高等学校科技计划项目(J12LN37);;青岛市科技计划项目(15-9-2-109-nsh); |
| |
摘 要: | 为解决当前工业机器人抓取检测任务中存在的精度不足的问题,提出了一种基于边缘提取与空洞卷积的抓取目标检测算法。该方法以当前较为稳定的YOLOv3网络作为主干网络,进行基于边缘提取的多通道特征融合,添加Edge Feature模块提高网络的检测精度和检测速度;基于空洞卷积技术优化自适应空间特征融合,形成Dilation-ASFF网络,替换原部分网络,提高网络对多尺度信息的感受能力以及网络检测精度;用CIoU的方法修改抓取框回归损失函数,提高网络的收敛能力与精度。改进后的算法在Cornell抓取检测数据集上的AP达到96.79%,检测速度达到0.063 s/样本,相比于原版YOLOv3网络,AP提高了2.98%。本改进算法在保证检测实时性的基础上,大大提高了检测精度,体现了研究的理论价值和应用价值。
|
关 键 词: | 抓取检测 YOLOv3 空洞卷积 边缘提取 CIoU |
|
|