ocr(2)-crnn-mxnet-chinese-text-recognition

ubuntu 16.04
GTX1080
Mxnet 最新
pytho 2.7

1. 安装Mxnet

Mxnet 按照参考官网，选择源码GPU安装。

2. 安装warp-ctc

安装warp-ctc

$ git clone https://github.com/baidu-research/warp-ctc.git
$ cd warp-ctc
$ mkdir build
$ cd build
$ cmake ../
$ make

(更详细内容请看官网)

3. 安装mxnet-warp-ctc

进入到mxnet根目录
$ cd mxnet
$ make clean
在make/config.mk 添加warp-ctc 路径
$ echo 'WARPCTC_PATH = $(HOME)/warp-ctc' >>config.mk
$ echo 'MXNET_PLUGINS += plugin/warpctc/warpctc.mk' >>config.mk
其中"$(HOME)" 为warp-ctc所在的根目录，如果路径不对会编译不通过

重新编译mxnet
$ make -j $(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda USE_CUDNN=1

重新编译python

如果找不到libwarpctc.so ,在warp-ctc/build中把libwarpctc.so 复制到/usr/lib/

1. 下载 crnn-mxnet-chinese-text-recognition 并训练

$ git https://github.com/diaomin/crnn-mxnet-chinese-text-recognition.git

数据集train.txt格式
 image_name1 label1_1 label1_2 label1_3...
如图片1.jpg 上面的标签是hello
train.txt：
1.jpg h e l l o

训练

$ python train.py 2>&1 | tee log.txt

参考文献
[1] https://mxnet.incubator.apache.org/install/index.html
[2]https://github.com/alues/Mxnet_Install_Script/blob/master/Install_Mxnet.sh
[3] https://github.com/baidu-research/warp-ctc
[4] https://github.com/diaomin/crnn-mxnet-chinese-text-recognition