Code for paper:
Improving Calibration and Out-of-Distribution Detection in Medical Image Segmentation with Convolutional Neural Networks
By:
Davood Karimi and Ali Gholipour
Submitted to IEEE TMI
If you have any questions about the code, please email [email protected].
files included:
data_prep.py read and prepare datasets train_model.py train CNNs ood_detection.py for detecting out-of-distribution data on the trained model
dk_aux.py, dk_model.py, and dk_seg.py include functions that are used by the other codes.