Very excited to share the second part soon, but I'm still waiting for the results of training the neural network which takes considerable time even on a Tesla K80 GPU. Stay tuned for the update!