This repository is created after a discussion on a paper which can be found here.
Thanks to Fuat Beser for his presentation on a group called Derin Ogrenme.
- Clone the repository and cd into the directory
- Run
pip install --editable .
to install the CLI - Run the script using
trainer --model {architecture-name}
- Current architectures are
- lenet_5
- mvgg_5
- mvgg_6
- mvgg_7
- mvgg_8
- mvgg_9
- You can always use trainer --help command to see the options available
##What's done in this repo?
- Training regular MNIST images with a simple CNN model
- Converting regular images into negative ones and training the model only on negative images dataset.
- Combining both regular and negative images and train a model on this combined dataset
##To-Do
- Improve the CLI.
- Simply find a way to fix the issue with the negative images! :)