Implementation of Learning Deep Features for Discriminative Localization(paper)
This implementation is a simplified version of original project. The network was made easy to understand.
This code implements only the main functions of the original project and visually checks their results.
For more detail, please refer to the original project.
- MNIST
- CIFAR10
dataset = 'CIFAR10'
img_size = 32
batch_size = 128
epoch = 5
learning_rate = 0.001
This repository is influenced by zhoubolei and KangBK0120