TensorFlow 2——替换【tensorflow.compat.v1.contrib.rnn.LSTMCell】解决方案

墨蓝 2023-01-12 11:48 182阅读 0赞

# 问题描述 #

> 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][http_www.bubuko.com_infodetail-3267863.html]

[https://blog.csdn.net/wo\_shanhaiguan/article/details/100674388][https_blog.csdn.net_wo_shanhaiguan_article_details_100674388]

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

[http_www.bubuko.com_infodetail-3267863.html]: http://www.bubuko.com/infodetail-3267863.html
[https_blog.csdn.net_wo_shanhaiguan_article_details_100674388]: https://blog.csdn.net/wo_shanhaiguan/article/details/100674388
[tf2.0_module _tensorflow_ has no attribute _contrib]: https://www.pianshen.com/article/99201848242/