It's really impressive work! Could you please let me know if there are training scripts available? I'd like to train it on my own dataset.