TensorFlow with tf.layers Implementation of Kaggle Cat vs Dog Challenge
Preparations
mkdir ckpt
mkdir graphs
mkdir batches
mkdir npys
- unzip data in
./data
pip install -r requirements.txt
- Install OpenCV 2.4.13 for Python
Steps
python utils/prepare_data.py
- To train Deep CNN Model, run:
python train_cnn.py DEEP
- Else, run:
python train_cnn.py SIMPLE