tha-santacruz / BayesianUNet
Pytorch Bayesian UNet model for segmentation and uncertainty prediction
☆23Updated 2 years ago
Alternatives and similar repositories for BayesianUNet:
Users that are interested in BayesianUNet are comparing it to the libraries listed below
- 3D Bayesian Convolutional Neural Network (BCNN) for Credible Geometric Uncertainty. Code for the paper: https://arxiv.org/abs/1910.10793☆62Updated last year
- ResUnetA implementation with pytorch example remote sensing image☆12Updated 4 years ago
- Joint Topology-preserving and Feature-refinement Network for Curvilinear Structure Segmentation (ICCV21)☆28Updated 3 years ago
- Migrate to PyTorch. Re-implementation of Bayesian Convolutional Neural Networks (BCNNs)☆59Updated 4 years ago
- Unet, ResUnet, DenseUnet, SE-ResUnet, Dual Attention ResUnet, Attention Gates ResUnet☆17Updated 3 years ago
- Neural Network for semantic segmentation☆66Updated 5 years ago
- A Framework for Systematic Validation of Uncertainty Estimation in Semantic Segmentation☆27Updated 9 months ago
- pytorch implementation of SEG-GRAD-CAM,which based on grad-cam☆51Updated 4 years ago
- 2d or 3d region growing segmentation on gray scale images☆28Updated 4 years ago
- ☆23Updated last year
- ☆13Updated 2 years ago
- This repo contains the 3D implementation of the commonly used attention mechanism for imaging.☆44Updated 2 years ago
- Compound loss for PyTorch☆45Updated 2 years ago
- ☆36Updated 2 years ago
- Optimization with JDTLoss and Evaluation with Fine-grained mIoUs for Semantic Segmentation☆51Updated 8 months ago
- ☆10Updated 2 years ago
- ☆39Updated 5 months ago
- Weak-Mamba-UNet: Visual Mamba Makes CNN and ViT Work Better for Scribble-based Medical Image Segmentation☆16Updated 9 months ago
- ☆9Updated last year
- A repository for semi supervised image segmentation using Mean Teacher☆29Updated 5 years ago
- ☆23Updated 7 months ago
- ☆22Updated last year
- Multi-task Attention-based Semi-supervised Learning framework for image segmentation☆42Updated 4 years ago
- using monte carlo dropout to have uncertainty estimation of predictions☆14Updated 5 years ago
- ComBiNet: Compact Convolutional Bayesian Neural Network for Image Segmentation☆18Updated last year
- ☆24Updated 3 years ago
- ☆59Updated last year
- ☆10Updated 2 years ago
- ☆34Updated 2 years ago
- ☆22Updated 4 years ago