TF-Slim(tensorflow.contrib.slim)是TensorFlow中的一个轻量级库,用来方便的定义、训练和评估模型,TF-Slim的代码可以和TensorFlow的其他代码自由组合。
参考资料:
- 源码及文档:https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/slim
- Slim模型仓库:https://github.com/tensorflow/models/tree/master/slim
组件
在slim的__init__.py文件可以看到所有的组件
arg_scope: 新增的域定义,带有更丰富的默认参数data: 读数据模块,包含数据工具data providers, parallel_reader, decodingqueues: 用于控制QueueRunner的上下文管理器variables: 更方便的操作变量layers: Layer模块,包含很多预定义的Layer,比Operator抽象层次更高losses: Loss函数learning: 训练模型evaluation: 评估模型metrics: 评估指标nets: 常见模型的实现