Hi,
Thanks for this repo, it's super awesome. Are you planning on releasing a pretrained model for this work?
Thanks!
Yes, the pre-trained model can be found at here.