SUE - Segmentation Uncertainty Estimation

Binary Segmentation Uncertainty Estimation in PyTorch.

Uncertainty Estimation Methods

The following methods are implemented in unc_estimation.py:

MCDO - Monte Carlo Dropout
TTA - Test Time Augmentations
Ensemble - Ensemble of models with similar architecture but different initialization

Name		Name	Last commit message	Last commit date
Latest commit History 18 Commits
.idea		.idea
models2D		models2D
README.md		README.md
main.py		main.py
main_utils.py		main_utils.py
run_main.sh		run_main.sh
unc_estimation.py		unc_estimation.py