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
即可。