NVlabs / DeepIM-PyTorch

PyTorch implementation of the DeepIM framework
Other
91 stars 19 forks source link

Can we use this repository to train on custom data? #5

Open eugval opened 3 years ago

eugval commented 3 years ago

Hello!

First of all, thank you for making this code available, it look awesome!

I have a custom dataset (including object models, background images, and region of possible camera poses on the object models frame) that I would like to train DeepIM on, would you say it would be easy to adapt this repository for this use-case?

If so, would it please be possible to give me any tips on where to start/ how to go about it?

Thanks! :)