Open-Set Object Detection Using 

Classification-free Object Proposal and Instance-level Contrastive Learning

Zhongxiang Zhou Yifei Yang Yue Wang Rong Xiong

State Key Laboratory of Industrial Control Technology and Institute of Cyber-Systems and Control

 Zhejiang University

[arXiv]              [Code]

Abstract

Detecting both known and unknown objects is a fundamental skill for robot manipulation in unstructured environments. Open-set object detection (OSOD) is a promising direction to handle the problem consisting of two subtasks: objects and background separation, and open-set object classification. In this paper, we present Openset RCNN to address the challenging OSOD. To disambiguate unknown objects and background in the first subtask, we propose to use classification-free region proposal network (CF-RPN) which estimates the objectness score of each region purely using cues from object's location and shape preventing overfitting to the training categories. To identify unknown objects in the second subtask, we propose to represent them using the complementary region of known categories in a latent space which is accomplished by a prototype learning network (PLN). PLN performs instance-level contrastive learning to encode proposals to a latent space and builds a compact region centering with a prototype for each known category. Further, we note that the detection performance of unknown objects can not be unbiasedly evaluated on the situation that commonly used object detection datasets are not fully annotated. Thus, a new benchmark is introduced by reorganizing GraspNet-1billion, a robotic grasp pose detection dataset with complete annotation. Extensive experiments demonstrate the merits of our method. We finally show that our Openset RCNN can endow the robot with an open-set perception ability to support robotic rearrangement tasks in cluttered environments.

Videos

Robotic rearrangement Task 5 - Case 1 using Openset RCNN

Robotic rearrangement Task 5 - Case 2 using Openset RCNN

Robotic rearrangement Task 5 - Case 3 using Openset RCNN

Robotic rearrangement Task 5 - Case 4 using Openset RCNN

Robotic rearrangement Task 5 - Case 5 using Openset RCNN

Robotic rearrangement Task 5 - Case 1 using OpenDet

Robotic rearrangement Task 5 - Case 2 using OpenDet

Robotic rearrangement Task 5 - Case 3 using OpenDet

Robotic rearrangement Task 5 - Case 4 using OpenDet

Robotic rearrangement Task 5 - Case 5 using OpenDet

Robotic rearrangement Task 5 - Case 1 using PROSER

Robotic rearrangement Task 5 - Case 2 using PROSER

Robotic rearrangement Task 5 - Case 3 using PROSER

Robotic rearrangement Task 5 - Case 4 using PROSER

Robotic rearrangement Task 5 - Case 5 using PROSER

Task 5 success rate

Number of Wrongly Detected Objects in Task 5

Results

Citation

If you want to cite our work, please use:

@article{zhou2023open,

  title={Open-Set Object Detection Using Classification-Free Object Proposal and Instance-Level Contrastive Learning},

  author={Zhou, Zhongxiang and Yang, Yifei and Wang, Yue and Xiong, Rong},

  journal={IEEE Robotics and Automation Letters},

  volume={8},

  number={3},

  pages={1691--1698},

  year={2023},

  publisher={IEEE}

}