This is the PyTorch implementation of Neural 3D Mesh Renderer v2. For the original code, please visit original repo
- Added PyTorch support
- Some bugfixes
- Some improvements in index & CUDA calls
- Some improvements in project structure & applied refactoring
- Maintained the original code under
neural_renderer_chainer
and tests undertests_chainer
- PyTorch tests are comparing results against the Chainer implementation.
- Added support for multiple GPUs
sudo python setup.py install