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(字母)是全方向识别。