由于需要用到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