图像处理方面的论文

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用于计算评估指标

相关文章
相关标签/搜索