本僅包含初級階段如何有效運行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(字母)是全方向識別。