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 的问题。