logo资料库

TensorFlow2没有contrib_slim包的问题.pdf

第1页 / 共1页
资料共1页,全文预览结束
Tensorflow2 没有 tf.contrib.slim 包的问题解决方法: Tensorflow2.0 版本修改之后有了很大的变动,其中 tf.contrib 被完全弃用了是 2.0 的一个重 大的变化,但 import tensorflow.contrib.slim as slim slim 作为一个高级封装,已经在很多之 前的版本中广泛使用(笔者就是在 tf2 上跑 faster rcnn 时需要解决该问题)。 搜索了很多次,大部分给的意见都是重新修改代码或者降低 tf 版本,没有找到解决途径 后来在 https://github.com/tensorflow/models/issues/8020 该网址看到 Tf-slim 有一个独立于 tensorflow 的镜像可以以 tf.compat.v1 兼容模式下使用,安装该 包即可。 提供两种安装方式: 1. https://github.com/google-research/tf-slim 该地址下载 zip 压缩包, 解压缩后 cd 到文件目录 Python setup.up install 安装 2. 也可以直接 pip install tf_slim 安装 导入方法: import tf_slim as slim 即可使用 slim 库,是的 tf2 可以在 tf.compat.v1 模式下运行 faster rcnn 代码 代码修改部分: 注释掉的部分为原代码,修改后可兼容运行,解决 tf2 中使用 slim 的问题。
分享到:
收藏