/keras-bda

A Bayesian Data Augmentation Approach for Learning Deep Models in Keras. Here is the link to a pytorch version: https://github.com/toantm/pytorch-bda

Primary LanguagePython

keras-bda

Train the three sub-networks in the paper "A Bayesian Data Augmentation Approach for Learning Deep Models" (NIPS 2017). The baseline classifiers are Lenet5, ResNet18, ResNetpa on the MNIST dataset . The code was adapted from: https://github.com/lukedeo/keras-acgan