Image classification + CAM with keras
- python 2.7
- OpenCV 3.2.0
- pandas
- numpy
- Keras 2.0.7
- TensorFlow 1.3.0
Input data(only for evaluation)
└── data
└── cats_vs_dogs
└── validation
└── cat
└── xxx.png (name doesn't matter)
└── dog
└── xxx.png (name doesn't matter)
The dataset directory structure is followed to use the Keras DataGen Framework.
checkpoint files,
└── checkpoint
└── (ckpt_name)
├── model.json
├── weight.xx.h5
└── ...
To test a model
$ python main.py --mode cam --ckpt_name weight --data_path ./data/cats_vs_dogs/validation --batch_size 10