TensorFlow 2——【module ‘tensorflow.compat.v1‘ has no attribute ‘contrib‘】解决方案

问题描述

Traceback (most recent call last):
  File "D:/Code/Project/a18/ocr/demo.py", line 16, in <module>
    import ocr
  File "D:\Code\Project\a18\ocr\ocr.py", line 18, in <module>
    from model.ctpn.text_detect import text_detect as text_detect0
  File "D:\Code\Project\a18\ocr\model\ctpn\text_detect.py", line 5, in <module>
    from model.ctpn.model import ctpn
  File "D:\Code\Project\a18\ocr\model\ctpn\model.py", line 48, in <module>
    sess, saver, net = load_tf_model()
  File "D:\Code\Project\a18\ocr\model\ctpn\model.py", line 28, in load_tf_model
    net = get_network("VGGnet_test")
  File "D:\Code\Project\a18\ocr\model\ctpn\lib\networks\factory.py", line 9, in get_network
    return VGGnet_test()
  File "D:\Code\Project\a18\ocr\model\ctpn\lib\networks\VGGnet_test.py", line 24, in __init__
    self.setup()
  File "D:\Code\Project\a18\ocr\model\ctpn\lib\networks\VGGnet_test.py", line 53, in setup
    (self.feed('rpn_conv/3x3').Bilstm(512, 128, 512, name='lstm_o'))
  File "D:\Code\Project\a18\ocr\model\ctpn\lib\networks\network.py", line 24, in layer_decorated
    layer_output = op(self, layer_input, *args, **kwargs)
  File "D:\Code\Project\a18\ocr\model\ctpn\lib\networks\network.py", line 102, in Bilstm
    lstm_bw_cell = tf.contrib.rnn.LSTMCell(d_h, state_is_tuple=True)
AttributeError: module 'tensorflow.compat.v1' has no attribute 'contrib'

问题分析

TensorFlow 2 删除了部分v1的库

解决方案

方法一：

退回1.x版本

方法二：

tensorflow.compat.v1.nn

可以替换部分 

参考文章

http://www.bubuko.com/infodetail-3267863.html

tf API 研读1：tf.nn，tf.layers， tf.contrib概述

tf2.0不降版本也能完美解决module ‘tensorflow’ has no attribute ‘contrib’的问题