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 上相关介绍,或参考其中文版翻译

  • 4
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值