Release for the paper submission

Separation to demo and training scripts.

To run the repository you only need Python 3.7 and NVIDIA docker. Follow the demo script to apply our neural networks to your own data.