How to apply this model on MRI data？

Hello, recently my job requires me to segment the ventricles on MRI images. After reviewing your work, I was extremely excited. However, my understanding of generative models is quite limited, so I hope to hear your suggestions so that I can carry out my work on the MRI dataset. It would be even better if you could provide a minimal code example.