search

MIC-DKFZ / nnUNet

Apache License 2.0
5.91k stars 1.76k forks source link

Ensemble #2468

Open Body123 opened 2 months ago

Body123 commented 2 months ago

I trained 2d image for segmentation task with 2D configuration and I write that command: nnUNetv2_find_best_configuration DATASET_NAME_OR_ID -c CONFIGURATIONS but it didn't give me a command to use ensemble how can I use ensemble for those 5 folds

Body123 commented 2 months ago

is their any help

Lars-Kraemer commented 2 months ago

Hey @Body123 ,

If you trained all 5 folds just use the nnUNetv2_predict command and nnUNet automatically ensembles these 5 folds.

nnUNetv2_predict -d DATASET_ID -i INPUT_FOLDER -o OUTPUT_FOLDER -c CONFIGURATION

Best, Lars