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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章