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