解决tensorflow2.x中使用tf.contrib.slim包时出现的No module named:tensorflow.contrib 问题

简介

Tensorflow2.x 版本较1.x版本有了很大的变动,以使TensorFlow用户更加高效。其中 tf.contrib 被完全弃用了是 2.x版本的一个重大的变化,但 import tensorflow.contrib.slim as slim slim 作为一个高级封装,已经在很多之前的版本中广泛使用。现在大部分的源码还是以tensorflow1.x版本为基础写的,这导致了一些已经在2.x版本中移除的模块无法使用。

主要问题

在运行import tensorflow.contrib.slim as slim
在这里插入图片描述

出现了:ModuleNotFoundError: No module named 'tensorflow.contrib 错误
在这里插入图片描述

解决方案

经查询现有的解决方案,大部分采用了降低版本的方法,如果想采用此方法可以自己去查询。
因为不想采用降低版本的方法进行解决,经过搜索在github中查询到此信息
链接: tf.contrib.slim is not worked in tensorflow 2.0 what is the alternative for that?.

Tf-slim 有一个独立于 tensorflow 的镜像可以以 tf.compat.v1 兼容模式下使用,安装该包即可Tf-slim 有一个独立于 tensorflow 的镜像可以以 tf.compat.v1 兼容模式下使用,安装该 包即可

TF-Slim是一个轻量级的库,用于在TensorFlow中定义,训练和评估复杂的模型。tf-slim的组件可以与本机tensorflow以及其他框架自由混合。
这里可以查询到关于Slim的信息: link.

操作

在cmd中使用pip下载TF-Slim

pip install --upgrade tf_slim

在这里插入图片描述

完成下载:
在这里插入图片描述

使用Slim库时注意:

#import tensorflow.contrib.slim as slim
import tf_slim as slim

其中,注释掉的部分是源代码,修改后不再报错。
(ps :能力有限,如有错误,敬请指出。

  • 39
    点赞
  • 87
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Fly蛋

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值