本仅包含初级阶段如何有效运行Demo,下篇文章将会介绍如何训练模型。
环境搭建
1.python3.6
2.tensorflow
3.CTPN源码
下载地址: https://github.com/eragonruan/text-detection-ctpn/releases
解压源码,一个是已经训练好的模型。
运行中的问题
使用PyCharm或者IDEA打开项目后,选择demo_pb.py运行将依次遇到的错误:
- easydict未安装 ,可以使用conda install easydict或者pip install easydict;
- 无法找到text.yml文件,修改路径ctpn/text.yml,去掉ctpn/;
- 无法找到ctpn.pb文件,将下载的训练模型移入ctpn/data目录下;
- 编码问题,对应修改cfg_key=cfg_key.encode(‘ascii’).decode(‘ascii’);
- 分隔符问题,对应修改,分割符要用’\’, 所以将split(’/’)改成split(’\’)就好了;
拓展
将text.yml中第36行中DETECT_MODE: H改成DETECT_MODE: O(字母)是全方向识别。