pytorch crnn训练

一.代码下载

代码
这个代码主要是针对中文的检测。

二.环境配置

WIN 10 or Ubuntu 16.04
PyTorch 1.2.0 (may fix ctc loss) with cuda 10.0 fire
yaml
easydict
tensorboardX
解释：pytorch版本不需要严格对应，但是建议安装新版本；pytorch版本>=1.1.0，其自带ctcloss,否则需要自己安装Warp-ctc。

三.测试demo

在根目录下运行demo.py即可，会输出相应文字，测试图片在images文件夹下的test.png图像。

四.训练

1.下载数据集数据
1）对数据进行解压，.rar文件在ubuntu中解压需要安装插件。

#压缩文件为rar的安装包
sudo apt-get install rar
#解压rar文件的安装包
sudo apt-get install unrar
#解压文件
sudo unrar x name.rar

备注：该文件名字Synthetic Chinese String Dataset .rar中包含空格，解压时建议自己修改以下名字。
解压后会生成一个images的文件夹，建议不要随便点进去，360万张图像容易卡电脑。
2）打开lib/config/360CC_config.yaml文件，将第15行，ROOT这个路径修改为自己解压生成的images文件路径

2.下载标签
1）label提取码：eaqb
2）将char_std_5990.txt，train.txt 和 test.txt文件放入lib/dataset/txt/中，txt文件夹下本身有这三个文件，删除原始的。
3.训练

python train.py --cfg lib/config/360CC_config.yaml