CNN用于张量流中的多类图像识别
CNN for multi-class image recognition in tensorflow
Notebook converted from Hvass-Labs’ tutorial in order to work with custom datasets, flexible image dimensions, 3-channel images, training over epochs, early stopping, and a deeper network. An updated version of the notebook for TensorFlow 2 is also included, along with a separate requirements file for that TensorFlow version.
This example uses Kaggle’s cats vs. dogs dataset.