Thank you for your sharing！ Is there a training code? I hope I can train my own dataset to see the effect

I am trying to reproduce your program. Is there any training code for this program? I only saw the pre-trained model