tensorflow_psenet如何跑起来

系统管理员 2022-10-06 02:55 168阅读 0赞

这个模型下载下来有一段时间了，主要是想用来看一下在某张照片上的效果，但是始终跑步起来。今天终于跑通了，纪念一下：

**一）首先下载地址：**  
[git clone https://github.com/liuheng92/tensorflow\_PSENet.git][git clone https_github.com_liuheng92_tensorflow_PSENet.git]  
模型下载地址：[百度云][Link 1]提取密码：pffd

模型下载下来之后，提取出来，不需要改动名称；

**二）pse.so文件的编译：**  
作者用的Python2环境，我用的是Python3，因此修改pse文件夹中Makefile中的内容如下：  
**主要改动Python改为Python3**  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NTg4NTIzMg_size_16_color_FFFFFF_t_70]  
**三）修改eval.py文件：**  
![在这里插入图片描述][20210610214802428.png]1）checkpoint\_path：我将模型解压到了model文件夹下，所以直接用了这个文件夹；  
2）代码172行：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NTg4NTIzMg_size_16_color_FFFFFF_t_70 1]将原来的注释掉，改为如下的内容：

with tf.Session(config=tf.ConfigProto(allow_soft_placement=True)) as sess:
                saver = tf.train.Saver()
                saver.restore(sess, 'model/model.ckpt')

3）xrange改为range

**四）修改utils\_tool.py文件**  
第12行，将其改为：

import queue

参考：  
1）[https://github.com/liuheng92/tensorflow\_PSENet][https_github.com_liuheng92_tensorflow_PSENet]  
2）[https://blog.csdn.net/liu506039293/article/details/102835275][https_blog.csdn.net_liu506039293_article_details_102835275]  
3）[https://ask.csdn.net/questions/5725078][https_ask.csdn.net_questions_5725078]

[git clone https_github.com_liuheng92_tensorflow_PSENet.git]: https://github.com/liuheng92/tensorflow_PSENet.git
[Link 1]: https://pan.baidu.com/s/14tQHf9MjuD0lSmwkoZhnCg
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NTg4NTIzMg_size_16_color_FFFFFF_t_70]: /images/20221005/27e1f43b86344d87bf0a7d74887f24a3.png
[20210610214802428.png]: /images/20221005/f870abf10e904048ad985119e49b2b71.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NTg4NTIzMg_size_16_color_FFFFFF_t_70 1]: /images/20221005/c276bacbc46c4f1fae57c4afe9fce540.png
[https_github.com_liuheng92_tensorflow_PSENet]: https://github.com/liuheng92/tensorflow_PSENet
[https_blog.csdn.net_liu506039293_article_details_102835275]: https://blog.csdn.net/liu506039293/article/details/102835275
[https_ask.csdn.net_questions_5725078]: https://ask.csdn.net/questions/5725078