基于CHINESE-OCR

ubuntu下部署CHINESE-OCR记录
1、github地址:https://github.com/xiaofengShi/CHINESE-OCR
系统环境ubuntu 18.04 LTS 自带python3.6.9
2、安装时需要参考同目录下的setup-python3.sh 等相关文件,除此之外还需要安装,安装时后面加上
-i https://mirrors.aliyun.com/pypi/simple/ 使用国内的源更快些。

pip3 install graphviz -i https://mirrors.aliyun.com/pypi/simple/
pip3 install pydot -i https://mirrors.aliyun.com/pypi/simple/
pip3 install torchvision -i https://mirrors.aliyun.com/pypi/simple/
pip3 install torch -i https://mirrors.aliyun.com/pypi/simple/
pip3 install utils -i https://mirrors.aliyun.com/pypi/simple/
pip3 install keras==2.2.5 -i https://mirrors.aliyun.com/pypi/simple/
pip3 install tensorflow==1.14.0 -i https://mirrors.aliyun.com/pypi/simple/
pip3 install numpy==1.16.0 -i https://mirrors.aliyun.com/pypi/simple/

刚开始安装keras时没注意选择版本,安装的最新的,它要求tensorflow2.2,于是就安装了,但是该框架tensorflow版本用的是1.7,所以里面有几处不适用的地方,干脆卸载的重装。numpy版本也不可太高,之前装的1.18.0 import tensorflow会出错,于是安装的1.16.0解决。

3、若出现以下错误

 cimport numpy as np
                ^
SyntaxError: invalid syntax

执行

cd ctpn/lib/utils/
./make-for-cpu.sh

4、接下来需要下载相应的模型文件,放到目录中,以便程序加载调用。
ctpn-checkpoint : https://pan.baidu.com/s/1aT-vHgq7nvLy4M_T6SwR1Q#list/path=%2F
CHINESE-OCR-master 的模型: 链接:https://pan.baidu.com/s/1WoIxpUrTxU1ELCV8fUiXag
提取码:xz4l (刷新链接时间 : 2020-6-30)
下面文件中,修改对应你实际的存放目录即可,前两个得需要绝对路径
ctpn/ctpn/model.py

ckpt_path = '/CHINESE-OCR/ctpn/ctpn/retrain/ckpt'

angle/predict.py

 model.load_weights('/CHINESE-OCR/angle/modelAngle.h5')

ocr/model.py

modelPath = os.path.join(os.getcwd(), "ocr/ocr0.2.h5")

5、接下来回到CHINESE-OCR 运行demo.py 即可。

你可能感兴趣的:(基于CHINESE-OCR)