一.代码下载
代码
这个代码主要是针对中文的检测。
二.环境配置
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