NVIDIA / DeepLearningExamples

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
13.31k stars 3.19k forks source link

Bug fix in Pytorch nnUNet preprocessor.py #1375

Open aecelaya opened 7 months ago

aecelaya commented 7 months ago

In PyTorch/Segmentation/nnUNet/data_preprocessing/preprocessor.py, line 220, you use the np.argmin function to find the low resolution axis for the target spacing variable. However, the low resolution axis is the position of the largest component of the target spacing. Hence, the np.argmax function is needed to find the low resolution axis.