Skip to content

Implementation of Deep-6DPose **This is not official implementation***

License

Notifications You must be signed in to change notification settings

ryohachiuma/Deep-6dPose

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep-6DPose: Recovering 6D Object Pose from a Single RGB Image

This is an implementation of Deep-6D pose on Python 3, Keras, and TensorFlow. The model generates bounding boxes, segmentation masks and 6 DoF pose for each instance of an object in the image. It's based on Feature Pyramid Network (FPN) and a ResNet101 backbone.

This is not a official implementation, and I forked and modified code from Mask R-CNN [https://github.com/matterport/Mask_RCNN]

I trained and tested with YCB Object Dataset [http://www.ycbbenchmarks.com/] To generate the dataset, run the script below.

cd samples/ycb
python3 generateData.py

To train the model, run the script below.

cd samples/ycb
python3 ycb.py train --dataset=/path/to/ycb/dataset --weights=coco

To test the model, run the script below.

cd samples/ycb
python3 ycb.py mask --dataset=/path/to/ycb/dataset --weights=last

I did not implement the evaluation code.

About

Implementation of Deep-6DPose **This is not official implementation***

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages