tensorflow_PSENet训练和测试

代码用的这个:https://github.com/liuheng92/tensorflow_PSENet

作者代码质量很好,根据他的步骤来就行。我把需要注意的地方写一下:

1.作者使用的Python2,python3需要修改的地方有2处

1.1 utils_tool.py12行:

import queue改成:

try:
    import queue
except ImportError:
    import Queue as queue

 1.2 eval.py 228行:

xrange改成range

 

2.g++版本不够的话,pse是不能编译。我是4.8版本的,所以要升级一下。同时,用python3的话,把pse/Makefile文件中的,

第一行:$(shell python-config --cflags)改成$(shell python3-config --cflags)

第二行:$(shell python-config --ldflags)改成$(shell python3-config --ldflags)

不要去下载源码编译,很浪费时间。https://www.jianshu.com/p/a54c882ac513通过这个blog去升级就行。

 

3.model下载下来之后没有checkpoint这个文件,自己新建一个:

model_checkpoint_path: "./model.ckpt"
all_model_checkpoint_paths: "./model.ckpt"

4.数据集下下来,放在data/icdar2015下面即可(自己创建这个文件夹)

 

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 16
    评论
评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值