HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Robot Instance Segmentation with Few Annotations for Grasping

Moshe Kimhi David Vainshtein Chaim Baskin Dotan Di Castro

Robot Instance Segmentation with Few Annotations for Grasping

Abstract

The ability of robots to manipulate objects relies heavily on their aptitude for visual perception. In domains characterized by cluttered scenes and high object variability, most methods call for vast labeled datasets, laboriously hand-annotated, with the aim of training capable models. Once deployed, the challenge of generalizing to unfamiliar objects implies that the model must evolve alongside its domain. To address this, we propose a novel framework that combines Semi-Supervised Learning (SSL) with Learning Through Interaction (LTI), allowing a model to learn by observing scene alterations and leverage visual consistency despite temporal gaps without requiring curated data of interaction sequences. As a result, our approach exploits partially annotated data through self-supervision and incorporates temporal context using pseudo-sequences generated from unlabeled still images. We validate our method on two common benchmarks, ARMBench mix-object-tote and OCID, where it achieves state-of-the-art performance. Notably, on ARMBench, we attain an $\text{AP}{50}$ of $86.37$, almost a $20\%$ improvement over existing work, and obtain remarkable results in scenarios with extremely low annotation, achieving an $\text{AP}{50}$ score of $84.89$ with just $1 \%$ of annotated data compared to $72$ presented in ARMBench on the fully annotated counterpart.

Code Repositories

mkimhi/RISE
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
instance-segmentation-on-armbenchRISE (R101)
AP50: 84.74
AP75: 75.93
instance-segmentation-on-armbenchDeformable DETR
AP50: 77.03
AP75: 63.4
instance-segmentation-on-armbenchRISE (R50)
AP50: 83.53
AP75: 75.15
instance-segmentation-on-armbenchMask2Former
AP50: 81.2
AP75: 74.0
instance-segmentation-on-armbenchRISE (VIT-B)
AP50: 86.37
AP75: 77.51

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp