Skip to content
master
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 

README.rst

Tensorflow "Bayesian U-Net" aka BUNet

This is the source code for the MICCAI 2018 Paper, Exploring Uncertainty Measures in Deep Networks for Multiple Sclerosis Lesion Detection and Segmentation (Nair et al.), of which I am the first author.

The network architecture is a heavily modified U-Net (Ronneberger et al.), developed in Tensorflow. The network is augmented to provide the following 4 different uncertainty measures as an output.

  1. Mutual Information (Gal et al.)
  2. Entropy (Gal et al.)
  3. MC Sample Variance (Leibig et al.)
  4. Predicted Variance (Kendall and Gal)

Details about the network architecture, and the equations for the uncertainty measures can be found in the paper here: https://arxiv.org/abs/1808.01200

The dataset used for this project comes from a large, proprietary, multi-site, multi-scanner, clinical MS dataset. As such, to use this code you will have to modify the dataprovider to be specific to your dataset.

Training:

  1. pip install -r requirements.txt
  2. python bunet_launcher.py -o ./path_to_output/ -c bunet/configs/train_bunet.json

Author: Tanya Nair

About

No description, website, or topics provided.

Resources

License

Releases

No releases published

Packages

No packages published

Languages