nnUNet_predict - nntrongnghia/TDSI21-Shoulder-Muscle-Segmentation GitHub Wiki
Preprocessing
- Preprocess images with
GenericPreprocessor.preprocess_test_case
for 3D andPreprocessorFor2D.preprocess_test_case
for 2D - Crop non-zero regions
- Transpose based on predefined plan generated by
nnUNet_plan_and_preprocess
- Resample:
- Normalize:
Predicting
- Run prediction of 5 folds and then calculate the mean of softmax values
- Save segmentation nifti from this mean softmax values
- For each fold:
- Prediction on patches with Test Time Augmentation - mirroring patches on all axes then get the mean values
- If more than 1 patches, use Gaussian importance map to merge patch prediction
Postprocessing
See postprocessing page