Environment setup
To train the deep CNN, you will need a computer with a CUDA-compatible GPU. I used an Ubuntu 16.x machine with NVidia GTX980 GPU for model training, and a macOS machine to convert the model to Core ML format. If you don't have CUDA-compatible GPU, you can try to train the model on CPU; but be aware that this will take a lot of time. Also, the trained model for this chapter is available in the supplementary materials, so if you prefer not to contribute to the global warming by retraining the model from scratch, it's also possible.
Here's a list of what should be installed on your system to train the network:
- Latest NVIDIA drivers
- CUDA 8.0
- cuDNN 5.1
- Python 2.7
tensorflow-gpu
(or TensorFlow for CPU-only mode)- Keras
- Keras-viz
- Matplotlib, Pandas
Please, refer to the official sites for the installation instructions.