Home

Awesome

Introduction

This is the toolbox for The YCB-Video dataset introduced for 6D object pose estimation. It provides accurate 6D poses of 21 objects from the YCB dataset observed in 92 videos with 133,827 frames.

License

The YCB-Video dataset is released under the MIT License (refer to the LICENSE file for details).

Citing

If you find our dataset useful in your research, please consider citing:

@article{xiang2017posecnn,
author    = {Xiang, Yu and Schmidt, Tanner and Narayanan, Venkatraman and Fox, Dieter},
title     = {PoseCNN: A Convolutional Neural Network for 6D Object Pose Estimation in Cluttered Scenes},
journal   = {arXiv preprint arXiv:1711.00199},
year      = {2017}
}

Annotation format

The *-meta.mat file in the YCB-Video dataset contains the following fields:

Usage

  1. Download the YCB-Video dataset from here.

  2. Set your path of the YCB-Video dataset in globals.m (required).

  3. show_pose_annotations.m displays the overlays of 3D shapes onto images according to our annotations. Check the code of this function to understand the annotation format.

  4. show_pose_results.m displays the 6D pose estimation results from PoseCNN. Unzip results_PoseCNN.zip before calling the function.

  5. evaluate_poses_stereo.m evaluates our results on the stereo pairs. Check the code of this function to understand the evaluation metric.

  6. evaluate_poses_keyframe.m evaluates our results on the keyframes.

  7. plot_accuracy_stereo.m plots all the accuracy-threshold curves from the stereo pairs.

  8. plot_accuracy_keyframe.m plots all the accuracy-threshold curves from the keyframes.