pytorch_demo.py

# coding:utf-8
import time
from glob import glob

import numpy as np
from PIL import Image

import pytorch_model as model
# ces

paths = glob('./test/*.*')

if __name__ == '__main__':
    # im = Image.open("./test/ttttt.png")
    im = Image.open("./test/ttttt.png")
    img = np.array(im.convert('RGB'))
    t = time.time()
    '''
    result,img,angel分别对应-识别结果，图像的数组，文字旋转角度
    '''
    result, img, angle = model.model(
        img, adjust=True, detectAngle=True)
    print("It takes time:{}s".format(time.time() - t))
    print("---------------------------------------")
    for key in result:
        print(result[key][1])