how to train on my own data

[artless-spirit]

Thank you for your open sorce job,but there are only inference code,can you tell me how to train the model on my own data ,Or can you provide the training code,thank you

[weiwei2021]

You can use FreeSurfer to process your data first to get the cortex surface ?h.white. And then, you can use the mris_volmask command to convert it into a levelset representation. The main part of the model is a 3D UNet-based regressor, you can try to implement it.