Bayesian U-Net for Keras

This is a Keras re-implementation for Bayesian U-Net

Currently, this repository only supports epistemic uncertainty...

Requirements

• Python 3
• CPU or NVIDIA GPU + CUDA CuDNN
• Keras and TensorFlow

Installation

• Install from this repository

git clone https://github.com/yuta-hi/keras_bayesian_unet
cd keras_bayesian_unet
pip install .

Usage

• Bayesian U-Net

# 2D
from keras_bcnn.models import BayesianUNet2D
input_shape = (512, 512, 1)
output_channles = 23
model = BayesianUNet2D(input_shape, output_channles).build()

# 3D
from keras_bcnn.models import BayesianUNet3D
input_shape = (128, 128, 128, 1)
output_channles = 23
model = BayesianUNet3D(input_shape, output_channles).build()

• MC sampler

from keras_bcnn.models import MCSampler
mc_iteration = 10
sampler = MCSampler(model, mc_iteration).build()

Related repositories

• bayesian_unet

• anatomy_viewer