Tensorflow 版本切换与 slim 问题

slim 概述

tf_slim 库能够使定义、训练和评估神经网络变得更简单。如其下的 arg_scope 和 layers 能够使定义模型变得更简单。

比如 slim.arg_scope，它可以给函数的参数自动赋予某些默认值。使用了之后，就不需要每次都重复设置参数，只需要在有需要修改的地方重新设置即可。如此，可以大大减少代码量，与此同时，增加可读性，减少误操作。当模型层数很多，同时有很多类似甚至完全一样的层时，使用 arg_scope 能大大减少定义模型的工作量。

导入 slim 模块

从 Tensorflow 2.0 开始，tf 移除了 tensorflow.contrib 库，其曾经 链接 也已失效，转而作为 tf_slim 模块单独下载使用。因此，在 tf2 版本中无法再使用 import tf.contrib.slim as slim 调用 slim 库。

下载：pip install --upgrade tf_slim
调用：import tf_slim as slim

关于 slim 下找不到 utils 的问题，请将 slim.utils 替换成 slim.layers.utils 即可。

slim 详情

可参考 github 上相关介绍，或参考其中文版翻译。