Git Product home page Git Product logo

category-6d-pose's Introduction

CRNet_IROS

Paper | Webpage

  • This is the official implementation of the paper: Category-Level 6D Object Pose Estimation via Cascaded Relation and Recurrent Reconstruction Networks (IROS 2021).
image

Quick Start

(1) Install Packages

Requires:

  • Python ≥ 3.6
  • Pytorch ≥ 1.0
  • CUDA ≥ 9.0

Compile nn_distance:

ROOT=/path/to/object-deformnet
cd $ROOT/lib/nn_distance
python setup.py install --user

(2) Dataset

Please download the Dataset following SPD.

Unzip and organize these files in $ROOT as follows:

data
├── CAMERA
│   ├── train
│   └── val
├── Real
│   ├── train
│   └── test
├── gts
│   ├── val
│   └── real_test
└── obj_models
    ├── train
    ├── val
    ├── real_train
    └── real_test

results
├── mrcnn_results
│   ├── real_test
│   └── val
└── nocs_results
    ├── real_test
    └── test

(3) Train and Evaluate

# train: python train.py

# test: python evaluate.py

Performance

Please download our models $ROOT/models.

Camera25:

Camera25 3D50 3D75 5°2cm 5°5cm 10°2cm 10°5cm
Our Paper 93.8 88.0 72.0 76.4 81.0 87.7
Released Model 93.7 88.1 71.2 76.5 81.1 88.4

Real275:

Real275 3D50 3D75 5°2cm 5°5cm 10°2cm 10°5cm
Our Paper 79.3 55.9 27.8 34.3 47.2 60.8
Released Model 78.9 57.8 31.5 34.9 51.5 62.8

Citation

If you find our work helpful, please consider citing:

@inproceedings{wang2021category,
  title={Category-level 6d object pose estimation via cascaded relation and recurrent reconstruction networks},
  author={Wang, Jiaze and Chen, Kai and Dou, Qi},
  booktitle={2021 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)},
  pages={4807--4814},
  year={2021},
  organization={IEEE}
}

You can also refer to our last work SGPA on ICCV2021:

@inproceedings{chen2021sgpa,
  title={Sgpa: Structure-guided prior adaptation for category-level 6d object pose estimation},
  author={Chen, Kai and Dou, Qi},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={2773--2782},
  year={2021}
}

Contact Us

WANG Jiaze: [email protected] | Homepage

Acknowledgment

Our implementation leverages the code from SPD, NOCS, 3PU and DCP.

category-6d-pose's People

Contributors

jiazewang avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.