Binary Segmentation Uncertainty Estimation in PyTorch.
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