yeerwen / UniSeg

MICCAI 2023 Paper (Early Acceptance)
Other
161 stars 5 forks source link

verse2020 merge dataset #8

Closed XFivezzz closed 1 year ago

XFivezzz commented 1 year ago

I am doing the task of verse 2020, and I have completed the Move splits The final. pkl step, when I merge with Python Each Sub Dataet.py, it seems to default to Task091 MOTS, how should I modify merge Each Sub Dataet.py makes it Task037_ VerSe20binary. Can I skip this step and train directly? image

yeerwen commented 1 year ago

Do you train only Ver2020 or do you need to follow our setup, i.e., joint training of 11 tasks? In the case of the former, you can use our Downstream code to fine-tune it, in the case of the latter, you need to follow the steps we provided to process the MOTS, VerSe20, BraTS21, Prastate, and AutoPET datasets in order, and then run the merge_each_sub_data.py file.

XFivezzz commented 1 year ago

I just want to train verse 2020, so what can I do directly?

yeerwen commented 1 year ago

If you are only training a task, you do not need to use our upstream task code. Instead, the downstream fine-tuning approach is more suitable because the benefits of pre-training can be enjoyed. Note that you need to rename the 'DoDNetPlans_plans_3D.pkl' and 'DoDNetData_plans_stage0' generated from the preprocessed VerSe dataset to 'nnUNetPlansv2.1_plans_3D.pkl' and 'nnUNetData_plans_v2.1_stage0', respectively, to avoid program errors.