Skip to content

knorth55/chainer-dense-fusion

Repository files navigation

chainer-dense-fusion

Build Status

This is Chainer implementation of DenseFusion: 6D Object Pose Estimation by Iterative Dense Fusion.

Original PyTorch repository is j96w/DenseFusion.

Requirement

Installation

We recommend to use Anacoda.

# Requirement installation
conda create -n dense-fusion python=3.6
source activate dense-fusion 
pip install opencv-python
pip install cupy

# Installation
git clone https://github.com/knorth55/chainer-dense-fusion.git
cd chainer-dense-fusion/
pip install -e .

Inference

cd examples/dense_fusion/
python demo.py --random

TODO

  • YCB Video Dataset
    • Add estimator inference script.
    • Add refiner inference script.
    • Add training script.
    • Reproduce original accuracy.

LICENSE

MIT LICENSE