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

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐