基于YOLOv4的抓取检测

摘要：

对于物品的纹理特征容易影响到抓取检测的精度,且原始的默认锚框不适合待抓取检测目标的尺度,以及在小目标物体检测效果不佳等问题,该文通过以下方法进行改进.首先通过图像处理将物品的原始图像经过处理使其只具有物品的轮廓信息,然后采用K-Means得到适合所检测目标的锚框尺度,以基于YOLOv4的神经网络模型建立改进后的YOLOv4抓取检测模型,最后去除YOLOv4中检测较大物体19×19的特征层检测层及其附近的卷积及池化层,以此降低系统复杂度减少参数,并且将第11层和第109层进行特征融合得到特征尺寸152×152使其能够更好地提取小目标特征.将原始图像和只具有轮廓信息的图像分别送入改进前和多种进行改进后的网络进行检测性能的分别对比.实验结果表明经过改进的网络抓取检测的平均成功率为81.5％,比原始的YOLOv4提高了4.3％,有效提升了抓取检测的精度并且加强了检测小目标的能力.

关键词：抓取检测;YOLOv4;轮廓提取;K-Means;小目标检测

所属期刊栏目：37

分类号：TP391.4(计算技术、计算机技术)

在线出版日期：2022-03-16（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：59-63,69