4 个月前

使用卷积神经网络进行实时抓取检测

使用卷积神经网络进行实时抓取检测

摘要

我们提出了一种基于卷积神经网络的精确实时机器人抓取检测方法。该网络通过单阶段回归直接预测可抓取的边界框,而无需使用传统的滑动窗口或区域提议技术。模型在性能上比现有最先进方法高出14个百分点,并且在GPU上可以达到每秒13帧的运行速度。我们的网络能够同时进行分类,从而在一个步骤中识别物体并找到一个合适的抓取矩形。对这一模型的改进版本通过使用局部约束预测机制来实现每个物体的多点抓取预测。这种局部约束模型的表现显著提升,尤其是在那些可以通过多种方式被抓取的物体上。

代码仓库

DucTranVan/grasp-detection-pytorch
pytorch
GitHub 中提及
hamed-hosseini/ggcnn
pytorch
GitHub 中提及

基准测试

基准方法指标
robotic-grasping-on-cornell-grasp-dataset-1AlexNet, MultiGrasp
5 fold cross validation: 88

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用卷积神经网络进行实时抓取检测 | 论文 | HyperAI超神经