darknet-ocr 是基于 darknet?框架，實現 CTPN 版本自然場景文字檢測與 CNN+CTCOCR 文字識別的項目。目前支持系統：mac/ubuntu python=3.6

實現功能

• [x] 文字方向檢測 0、90、180、270度檢測（支持dnn/tensorflow）
• [x] 支持(darknet/opencv dnn /keras)文字檢測,支持darknet/keras訓練
• [x] 不定長OCR訓練(英文、中英文) crnn\dense ocr 識別及訓練 ,新增pytorch轉keras模型代碼(tools/pytorch_to_keras.py)
• [x] 支持darknet 轉keras, keras轉darknet, pytorch 轉keras模型
• [x] 身份證/火車票結構化數據識別
• [x] 新增CNN+ctc模型，支持DNN模塊調用OCR，單行圖像平均時間為0.02秒以下
• [ ] CPU版本加速
• [ ] 支持基于用戶字典OCR識別
• [ ] 新增語言模型修正OCR識別結果
• [ ] 支持樹莓派實時識別方案

ocr 訓練數據集(壓縮包解碼：chineseocr)

百度網盤鏈接：https://pan.baidu.com/s/1UcUKUUELLwdM29zfbztzdw?提取碼: atwn

gofile 地址：http://gofile.me/4Nlqh/uT32hAjbx?密碼?https://github.com/chineseocr/chineseocr

環境部署

GPU部署 參考:setup.md
CPU部署 參考:setup-cpu.md

下載編譯 darknet

git clone https://github.com/pjreddie/darknet.git 
mv darknet chineseocr/
##編譯對GPU、cudnn的支持 修改 Makefile
#GPU=1
#CUDNN=1
#OPENCV=0
#OPENMP=0
make 

修改 darknet/python/darknet.py line 48
root = '/root/'##chineseocr所在目錄
lib = CDLL(root+"chineseocr/darknet/libdarknet.so", RTLD_GLOBAL)

下載模型文件

模型文件地址:

• 百度網盤:https://pan.baidu.com/s/1gTW9gwJR6hlwTuyB6nCkzQ
  other-links:http://gofile.me/4Nlqh/fNHlWzVWo
  復制文件夾中的所有文件到models目錄

模型轉換（非必須）

pytorch ocr 轉 keras ocr

python tools/pytorch_to_keras.py  -weights_path models/ocr-dense.pth -output_path models/ocr-dense-keras.h5

darknet 轉 keras

python tools/darknet_to_keras.py -cfg_path models/text.cfg -weights_path models/text.weights -output_path models/text.h5

keras 轉 darknet

python tools/keras_to_darknet.py -cfg_path models/text.cfg -weights_path models/text.h5 -output_path models/text.weights

模型選擇

參考config.py文件

構建 Docker 鏡像

##下載Anaconda3 python 環境安裝包（https://repo.anaconda.com/archive/Anaconda3-2019.03-Linux-x86_64.sh） 放置在chineseocr目錄下   
##建立鏡像   
docker build -t chineseocr .   
##啟動服務   
docker run -d -p 8080:8080 chineseocr /root/anaconda3/bin/python app.py

Web 服務啟動

cd chineseocr## 進入chineseocr目錄
python app.py 8080 ##8080端口號，可以設置任意端口

訪問服務

http://127.0.0.1:8080/ocr