TF-Slim网络
TF-Slim是tensorflow中定义、训练和评估复杂模型的轻量级库。tf-slim中的组件能够轻易地和原生tensorflow框架以及例如tf.contrib.learn这样的框架进行整合。框架
在使用时先导入库:函数
import tensorflow.contrib.slim as slim
优点:spa
一、代码更紧凑code
二、封装了不少流行模型,如vgg等,能够当作黑盒用,减小代码错误blog
训练过程:class
训练一个tensorflow模型,须要一个网络模型,一个损失函数,梯度计算方式和用于迭代计算模型权重的训练过程。TF-Slim提供了损失函数,同时也提供了一批运行训练和评估模型的帮助函数import
微调tensorflow
经过tf.train.Saver()从checkpoint中加载模型im
当咱们想把已训练好的模型用在不一样的数据集上时,能够恢复除全链接层之外的全部层。
评估
slim.metrics用于计算评估指标