手动将 tf.contrib.slim 升级到 tf 2.0

Question

我的 python 代码有问题，它使用了一些tf.contrib.slim功能，在升级到 tensorflow 到 2.0 后不再工作。

如何将以下内容升级到 tf 2.0：

import tensorflow.contrib.slim as slim
import tensorflow.contrib.slim.nets

# ...

net = slim.conv2d(
    inp, 
    dim,
    [3, 3],
    rate=1,
    normalizer_fn=slim.layer_norm,
    activation_fn=lrelu,
    scope='g_' + str(width) + '_conv1') 

谢谢。

Answer 1

tf-slim首先安装：

pip install --upgrade tf_slim

那么import tensorflow.contrib.slim as slim你需要写：

import tf_slim as slim。这解决了我同样的问题。希望这可以帮助。

Answer 2

您可以改用tf_slim包（https://github.com/google-research/tf-slim）

但是，将包更新到 TF2 的推送请求仍然未决。所以你应该从分支安装 pip install git+https://github.com/adrianc-a/tf-slim.git@remove_contrib