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

  • Post author:
  • Post category:其他


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



版权声明:本文为weixin_45885232原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。