TF-slim是用于定义,训练和评估复杂模型的TensorFlow(tensorflow.contrib.slim)的新型轻量级高级API。 该目录包含用于训练和评估使用TF-slim的几种广泛使用的卷积神经网络(CNN)图像分类模型的代码。 它包含脚本,允许您从头开始训练模型或从预培训的网络权重微调它们。 它还包含用于下载标准图像数据集的代码,将其转换为TensorFlow的本机TFRecord格式,并使用TF-Slim的数据读取和排队实用程序进行读取。
您可以轻松地对任何这些数据集进行任何模型的训练,如下所示。 我们还包括一个jupyter笔记本,它提供了如何使用TF-Slim进行图像分类的工作示例。