• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    公众号

HLIG/HUAWEIOCR-2019: DataFountain HUAWEI OCR 2019 rank Top1% ,Text Localization ...

原作者: [db:作者] 来自: 网络 收藏 邀请

开源软件名称(OpenSource Name):

HLIG/HUAWEIOCR-2019

开源软件地址(OpenSource Url):

https://github.com/HLIG/HUAWEIOCR-2019

开源编程语言(OpenSource Language):

Python 96.5%

开源软件介绍(OpenSource Introduction):

文字检测主要使用了浙大的pixellink

https://github.com/ZJULearning/pixel_link

在文字识别OCR模型上使用西安交通大学人工智能实践大赛第一名@ yinchangchang 的方案https://github.com/yinchangchang/ocr_densenet

针对本比赛的数据集,对代码进行了修改,改进的细节见代码和参考技术报告

https://discussion.datafountain.cn/questions/1571/answers/22399

安装步骤

1、首先代码的文件分为四个部分:

文字检测文件夹:detection

文字识别文件夹:recognition

存放测试图片的文件夹:test_images

最后结果和中间生成文件:intermediate_file

注意:存放文件夹的路径不能带中文!!!!!!!

2、安装步骤

所有的代码语言均为python2,系统环境为linux64

建议使用anaconda分别创建 识别检测的环境

解压到文件夹,假设为unzip_path

首先,安装文字检测环境

打开终端,进入HUAWEI_OCR2019/detection,安装检测的环境并激活,设该终端为文字检测终端

conda env create --file detection.txt
source activate pixel_link

然后,安装文字识别环境

新打开一个终端,进入HUAWEI_OCR2019/recognition

进入HUAWEI_OCR2019/detection安装检测的环境并激活,设该终端为文字识别终端

conda create --name ocr_densenet python=2.7
source activate ocr_densenet
pip install recognition.txt

从百度云下载文字检测网络权重文件,放到路径HUAWEI_OCR2019\detection\pixel_link\checkpoint下

从百度云下载文字识别网络权重文件,放到路径HUAWEI_OCR2019\recognition\densenet\model下

https://pan.baidu.com/s/1LVbfVa5ieibFg_X6iIuWfw

提取码:e8rd

至此,文字检测和文字识别的环境都装好了。

3、测试及相关命令

在创建了检测文字检测环境pixel_link,和文字识别网络环境ocr_densenet后

首先在文字检测终端!!!!!!!!,进入解压后的文件夹 cd unzip_path/HUAWEI_OCR2019

  1. 把网站上下载的测试图片集中的图片放到unzip_path/ HUAWEI_OCR2019/test_images

  2. 执行python test_to_padding.py 功能:将测试图片填充,并将结果输出到unzip_path/ HUAWEI_OCR2019/intermediate_file/padding_images_to_detection

  3. cd unzip_path/HUAWEI_OCR2019/detection/pixel_link 执行脚本./scripts/test_any.sh 3 checkpoint/model.ckpt-146343 ../../intermediate_file/padding_images_to_detection,运行检测的pixel link网络 功能:检测网络,将每一张填充后的图片的检测框信息输出到文件夹unzip_path/HUAWEI_OCR2019/intermediate_file/detect_result/txt中,将可视化的检测结果图片输出到unzip_path/HUAWEI_OCR2019/intermediate_file/detect_result/images中。

  4. cd unzip_path/HUAWEI_OCR2019返回主目录 双方案,B榜提交是方案1,但是更优秀的方案是2(方案2用了仿射变换,可以对倾斜的文字进行分割,由于时间仓促没用上)。

    1. (二选一)执行脚本python crop_and_reshape.py 功能:将填充后的测试图片根据检测结果分割为子图片,并输出到unzip_path/HUAWEI_OCR2019/intermediate_file/images_to_recognition文件夹下。
    2. (二选一)执行脚本python target_cut_test_many.py 功能:根据检测结果,将填充后的图片进行仿射变换后的分割,同样输出到unzip_path/HUAWEI_OCR2019/intermediate_file/images_to_recognition文件夹下。
  5. cd unzip_path/HUAWEI_OCR2019/recognition/densenet/code 在**文字识别终端!!!!!!!**切换到识别网络程序所在目录 执行脚本python main.py 功能:对切割后的图片进行神经网络识别,输出中间结果result.csvunzip_path/HUAWEI_OCR2019/intermediate_file/recognize_result

  6. 切换回文字检测终端!!!!!!!!! cd unzip_path/HUAWEI_OCR2019返回主目录 执行python txt_to_csv.py 功能:将检测结果的所有.txt文件转换为padding_detect_result.csv,并将结果输出到unzip_path/HUAWEI_OCR2019/intermediate_file/detect_result/csv文件夹下

  7. 执行python padding2unpadding_box.py 功能:将检测结果padding_detect_result.csv转换回未填充的结果unpadding_detect_result.csv,并将结果输出到unzip_path/HUAWEI_OCR2019/intermediate_file/detect_result/csv文件夹下

  8. 执行python get_predict_csv.py 功能:将未填充检测结果unpadding_detect_result.csv和识别结果result.csv文件组合为一个可提交的predict.csv文件,输出最终结果到unzip_path/HUAWEI_OCR2019/intermediate_file/final_result/文件夹下

unzip_path/HUAWEI_OCR2019/intermediate_file/final_result文件夹下即为最终提交的predict.csv结果

检测效果图

1555482830163

1555482850757

识别效果图

1555482792330




鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap