問題描述

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'

問題分析

TensorFlow2.x 刪除了部分v1的庫

解決方案

方法一：

退回1.x版本

方法二：

tensorflow.compat.v1.nn.rnn_cell.rnn.LSTMCell(d_h, state_is_tuple=True)

?

參考文章

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

https://blog.csdn.net/wo_shanhaiguan/article/details/100674388

tf2.0不降版本也能完美解決module ‘tensorflow’ has no attribute ‘contrib’的問題

?

總結

以上是生活随笔為你收集整理的TensorFlow 2——替换【tensorflow.compat.v1.contrib.rnn.LSTMCell】解决方案的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。