Python-基于Tensorflow和Keras实现端到端的不定长中文字符检测和识别

#-*- coding:utf-8 -*-
import os
import ocr
import time
import shutil
import numpy as np
from PIL import Image
from glob import glob
image_files = glob（‘./test_images/*.*‘）


if __name__ == ‘__main__‘:
    result_dir = ‘./test_result‘
    if os.path.exists（result_dir）:
        shutil.rmtree（result_dir）
    os.mkdir（result_dir）

    for image_file in sorted（image_files）:
        image = np.array（Image.open（image_file）.convert（‘RGB‘））
        t = time.time（）
        result image_framed = ocr.model（image）
        output_file = os.path.join（result_dir image_file.split（‘/‘）[-1]）
        Image.fromarray（image_framed）.save（output_file）
        print（“Mission complete it took {:.3f}s“.format（time.time（） - t））
        print（“
Recognition Result:
“）
        for key in result:
            print（result[key][1]）

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2018-10-08 03:37  chinese_ocr-master
     文件         148  2018-10-08 03:37  chinese_ocr-master.gitignore
     文件       11357  2018-10-08 03:37  chinese_ocr-masterLICENSE
     文件        1962  2018-10-08 03:37  chinese_ocr-masterREADME.md
     目录           0  2018-10-08 03:37  chinese_ocr-masterctpn
     文件        3958  2018-10-08 03:37  chinese_ocr-masterctpnREADME.md
     文件           0  2018-10-08 03:37  chinese_ocr-masterctpn\__init__.py
     目录           0  2018-10-08 03:37  chinese_ocr-masterctpncheckpoints
     文件    71572976  2018-10-08 03:37  chinese_ocr-masterctpncheckpointsVGGnet_fast_rcnn_iter_50000.ckpt.data-00000-of-00001
     文件        1643  2018-10-08 03:37  chinese_ocr-masterctpncheckpointsVGGnet_fast_rcnn_iter_50000.ckpt.index
     文件      649342  2018-10-08 03:37  chinese_ocr-masterctpncheckpointsVGGnet_fast_rcnn_iter_50000.ckpt.meta
     文件         121  2018-10-08 03:37  chinese_ocr-masterctpncheckpointscheckpoint
     目录           0  2018-10-08 03:37  chinese_ocr-masterctpnctpn
     文件        3994  2018-10-08 03:37  chinese_ocr-masterctpnctpndemo.py
     文件         872  2018-10-08 03:37  chinese_ocr-masterctpnctpn ext.yml
     文件        1197  2018-10-08 03:37  chinese_ocr-masterctpnctpn rain_net.py
     目录           0  2018-10-08 03:37  chinese_ocr-masterctpndata
     文件           9  2018-10-08 03:37  chinese_ocr-masterctpndataVOCdevkit2007
     目录           0  2018-10-08 03:37  chinese_ocr-masterctpndatademo
     文件      160420  2018-10-08 03:37  chinese_ocr-masterctpndatademo01.jpg
     文件       23022  2018-10-08 03:37  chinese_ocr-masterctpndatademo02.jpg
     文件      178066  2018-10-08 03:37  chinese_ocr-masterctpndatademo03.jpg
     文件      265586  2018-10-08 03:37  chinese_ocr-masterctpndatademo04.jpg
     文件      707796  2018-10-08 03:37  chinese_ocr-masterctpndatademo05.jpg
     文件      470595  2018-10-08 03:37  chinese_ocr-masterctpndatademo06.jpg
     文件       79041  2018-10-08 03:37  chinese_ocr-masterctpndatademo07.jpg
     文件      123596  2018-10-08 03:37  chinese_ocr-masterctpndatademo08.jpg
     文件      362936  2018-10-08 03:37  chinese_ocr-masterctpndatademo09.jpg
     文件       12952  2018-10-08 03:37  chinese_ocr-masterctpndatademo10.png
     目录           0  2018-10-08 03:37  chinese_ocr-masterctpndataoriented_results
     文件      210939  2018-10-08 03:37  chinese_ocr-masterctpndataoriented_results01.jpg
............此处省略96个文件信息