uni-medical/STU-Net

How to train from scratch(re-pretraining)?

eclipse0922 opened this issue · 1 comments

Thank you for sharing awesome work!

I did fine tuning for my dataset with pre-trained weight(small and base model)
It shows me great result.

Can you share the training script used to train the TotalsSegmentator dataset?
Or is it sufficient to use the method in this link?
https://github.com/wasserth/TotalSegmentator/blob/master/resources/train_nnunet.md

I would like to add more datasets to the pre-training model to maximise performance.

like Verse and CTPelvic1K or more,
Of course, I may have to change the information on their labels.

same question.
#11