3 个月前

基于RGB图像的机器人抓取检测与语义分割的端到端可训练深度神经网络

基于RGB图像的机器人抓取检测与语义分割的端到端可训练深度神经网络

摘要

在本研究中,我们提出了一种新型的端到端可训练的基于卷积神经网络(CNN)的架构,能够为平行板夹爪(parallel-plate gripper)的抓取检测以及语义分割任务提供高质量的输出结果。基于该架构,我们进一步设计了一种新颖的优化模块,该模块充分利用先前计算得到的抓取检测与语义分割结果,进一步提升了抓取检测的精度。所提出的网络在两个主流抓取检测数据集——Cornell与Jacquard上均达到了当前最先进的准确率。作为额外贡献,我们对OCID数据集提出了新的扩展版本,使其能够用于评估在极具挑战性场景下的抓取检测性能。利用该扩展数据集,我们验证了语义分割还可用于将抓取候选点分配至具体的物体类别,从而实现对场景中特定物体的精准抓取。

代码仓库

stefan-ainetter/grasp_det_seg_cnn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
robotic-grasping-on-cornell-grasp-dataset-1grasp_det_seg_cnn (rgb only, IW split)
5 fold cross validation: 98.2
robotic-grasping-on-jacquard-datasetgrasp_det_seg_cnn (rgb only)
Accuracy (%): 92.95

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于RGB图像的机器人抓取检测与语义分割的端到端可训练深度神经网络 | 论文 | HyperAI超神经