天天看点

TensorFlow1.15后版本没有tensorflow.contrib的解决办法

由于需要用到rcnn和slim这两个库,但是后面的版本中没有了,降低版本虽然是一个好办法,但可能会有其他的问题。

# from tensorflow.contrib import rnn
替换为:
rnn = tf.compat.v1.nn.rnn_cell
# from tensorflow.contrib import slim
替换为:
import tf_slim as slim
           

其中tf_slim需要安装:

pip install tf_slim
           

继续阅读