How to migrate to my model？Looking forward to your answer, thanks.

[Pantherk1]

Thank you for your excellent work, Dr. Zhou. I am a novice postgraduate, and my coding ability is not high. I want to use your method on my own two-class segmentation model. Which py files should I read or what should I pay attention to? Looking forward to your answer, thank you very much.