Hi,
Thanks for the wonderful work.

The code seems to have codes only for training the network.
Do you have any plan to release any source code for demonstrations?
I tried to change the code by myself. But it doesn't work correctly with the pre-trained model.