Skip to content

Latest commit

 

History

History
29 lines (25 loc) · 962 Bytes

train_nnUNet.md

File metadata and controls

29 lines (25 loc) · 962 Bytes
  1. Installation: https://github.com/MIC-DKFZ/nnUNet/blob/master/documentation/installation_instructions.md
  2. Dataset conversion: https://github.com/MIC-DKFZ/nnUNet/blob/master/nnunetv2/dataset_conversion/Dataset027_ACDC.py#L6
  3. Verify the integrity of data https://github.com/MIC-DKFZ/nnUNet/blob/master/documentation/how_to_use_nnunet.md#model-training
nnUNet_raw="../../../Dataset/nnUNet_raw" nnUNet_preprocessed="../../../Dataset/nnUNet_preprocessed" nnUNet_results="../../../Dataset/nnUNet_results" nnUNetv2_plan_and_preprocess -d 1 --verify_dataset_integrity
  1. Perform training
nnUNet_raw="../../../Dataset/nnUNet_raw" nnUNet_preprocessed="../../../Dataset/nnUNet_preprocessed" nnUNet_results="../../../Dataset/nnUNet_results"  CUDA_VISIBLE_DEVICES=0 nnUNetv2_train Dataset001_CAMUS 2d 0 --npz
  1. Inference on 1 fold

  1. FInd the best configuration

  1. Inference on all folds

  1. Perform post-processing