Implementation of CIFAR-10 classifier with TensorFlow
- Python 2.7
- Tensorflow
-
Convert to TFRecords format
$ python convert_to_tfrecords.py --data_dir ./data
-
(Optional) Test for reading TFRecords files
Open `read_tfrecords_sample.ipynb` in Jupyter
-
Train
$ python train.py --data_dir ./data --train_logdir ./logs/train
-
Retrain if you need
$ python train.py --data_dir ./data --train_logdir ./logs/train2 --restore_checkpoint ./logs/train/latest.ckpt
-
Evaluate
$ python eval.py --data_dir ./data --checkpoint_dir ./logs/train --eval_logdir ./logs/eval
-
Visualize
$ tensorboard --logdir ./logs
-
(Optional) Try to make an inference
Open `inference_sample.ipynb` in Jupyter
-
Clean
$ rm -rf ./logs or $ rm -rf ./logs/train2 or $ rm -rf ./logs/eval