Code utilized in Kristoffer Wickstrom's master thesis

This repository containts the for the master thesis entitled "Uncertainty Modelling and Interpretability in Convolutional Neural Networks for Polyp Segmentation"

GB Models

Contain the fully trained models with modified backward pass for Guided Backpropagation

Misc

Contain the scripts for creating uncertainty maps, interpretability maps and gradient uncertainty maps. Also includes scrips for mutual information estimation of DNNs

Models

Contain the code for FCN-32, FCN-16, FCN-8, SegNet and U-Net

Utility

Assortment of functions like data loader, training/validation/test and data augmentation.