天天看點

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
           

繼續閱讀