Nice work! The example results sound promising. It would be better if you could provide some pretrained models.