问题描述
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’的问题