/pytorch_bayesian_unet

Migrate to PyTorch. Re-implementation of Bayesian Convolutional Neural Networks (BCNNs)

Primary LanguagePythonMIT LicenseMIT

Stargazers