yangheng111 / darknet-ocr

darknet text detect and darknet cnn ocr

本项目基于darknet(https://github.com/pjreddie/darknet.git)框架实现CTPN版本自然场景文字检测 与CNN+CTCOCR文字识别

实现功能

• CPU版本最短边608时，检测速度小于1秒；
• 支持GPU
• 支持darknet直接训练CTPN（整理中）；
• 支持darknet直接训练CNN+CTC ocr（整理中）;

下载text.weights模型文件

模型文件地址:

• baidu pan

拷贝text.weights文件到models目录

编译对GPU的支持

sh make-gpu.sh

web服务启动(支持文件上传及URL图像)

cd darknet-ocr
python3 app.py 8080

访问服务

http://127.0.0.1:8080/text

识别结果展示

参考

1. yolo3 https://github.com/pjreddie/darknet.git
2. ctpn https://github.com/eragonruan/text-detection-ctpn
3. CTPN https://github.com/tianzhi0549/CTPN