TensorFlow CTPN文字定位与识别


本仅包含初级阶段如何有效运行Demo,下篇文章将会介绍如何训练模型。

环境搭建

1.python3.6
2.tensorflow
3.CTPN源码
下载地址: https://github.com/eragonruan/text-detection-ctpn/releases
在这里插入图片描述
解压源码,一个是已经训练好的模型。

运行中的问题

使用PyCharm或者IDEA打开项目后,选择demo_pb.py运行将依次遇到的错误:

  1. easydict未安装 ,可以使用conda install easydict或者pip install easydict;
  2. 无法找到text.yml文件,修改路径ctpn/text.yml,去掉ctpn/;
  3. 无法找到ctpn.pb文件,将下载的训练模型移入ctpn/data目录下;
  4. 编码问题,对应修改cfg_key=cfg_key.encode(‘ascii’).decode(‘ascii’);
  5. 分隔符问题,对应修改,分割符要用’\’, 所以将split(’/’)改成split(’\’)就好了;

拓展

将text.yml中第36行中DETECT_MODE: H改成DETECT_MODE: O(字母)是全方向识别。

猜你喜欢

转载自blog.csdn.net/baidu_33512336/article/details/105694777