Open noehsueh opened 3 months ago
Hi @noehsueh , have you already tried following the documentation here: https://github.com/MIC-DKFZ/nnUNet/blob/master/documentation/pretraining_and_finetuning.md
Let me know if that clarifies the use of overwrite_plans!
Hi @GregorKoehler,
Thanks for your reply. I'm using nnUNet v1. The documentation seems to work only for v2, but if I understood it correctly, the plan files contains the network topology, data processing, etc. it also requires training on the source dataset. As my source dataset is BraTS 2020 and I've already got the weights, I was wondering if there's a way of fine-tune the model without training nnUNet on Task082_BraTS2020 again. I ran this command to check that my data matches the ones of BraTS2020: nnUNet_plan_and_preprocess -t 082 --verify_dataset_integrity
. But didn't transfer any plans files between the source and the target. Just wondering if this was the right approach.
Thanks All the best, Noe
Hey Noe, it's very hard for us to debug old nnU-Net V1 issues when v2 has been out for a long time. Please try using that and it will be a lot easier to help you. We have also improved the documentation a lot. Best, Fabian
Hi Fabian,
I've been using nnUNet v1 as the pertained weights I found for BraTS 2020 were only available for this version. I would like to know if pertained models weights will be available for v2. Unfortunately, I have limited computational resources and might not be able to train the model from scratch.
Best, Noe
Hi @noehsueh, I'm looking into making pretrained weights available for BraTS 2020 in v2 and will post an update here soon. I'm certain we should be able to provide pretrained weights for this dataset.
Best, Gregor
Hi Gregor,
That would be very helpful, thanks!
Best Noe
Hi,
I'm trying to finetune nnUNet for a brain tumor segmentation task using pretrained weights from Task082_BraTS2020. My data consists of brain MRI images with 4 input modalities, and the labels match those used in Brats. I have followed the instructions in issue #774, but my results are unsatisfactory. I suspect there may be errors in my data preprocessing. Here's the code I used after exporting the environmental variables (I've replaced the Brats data with my custom data intended for finetuning in
nnUNet_raw_data_base/nnUNet_raw_data/Task082_BraTS2020
) and downloaded the pretrained model:Then, for all the folds (0, 1, 2, 3, 4), I ran the following command:
Where
nnUNetTrainerV2_250epochs
is a custom trainer I created to finetune the model.I haven't overwritten the plans because I thought that it would just preprocess the files using the Task082 plans. Should I have used the
-overwrite_plans
command instead? If so, what should the-overwrite_plans_identifier
be?Many thanks! Noe