小白學(xué)Python之實(shí)現(xiàn)OCR識(shí)別

	root = Tk() 
	root.title("小新的OCR")
	# 創(chuàng)建一個(gè)頂級(jí)菜單
    menubar = Menu(root)
    # 創(chuàng)建一個(gè)下拉菜單“文件”，然后將它添加到頂級(jí)菜單中
    filemenu = Menu(menubar, tearoff=False)
    filemenu.add_command(label="OCR", command=buttonCaptureClick, accelerator='Ctrl+N')
    filemenu.add_command(label="幫助",command=helpClick)
    filemenu.add_command(label="退出", command=root.quit)
    menubar.add_cascade(label="操作", menu=filemenu)
    # 顯示菜單
    root.config(menu=menubar)
    root.bind_all("Control-d>", lambda event: buttonCaptureClick())
    #啟動(dòng)消息主循環(huán)
    root.mainloop()  

#用來(lái)顯示全屏幕截圖并響應(yīng)二次截圖的窗口類
class MyCapture:
    def __init__(self, png):
        #變量X和Y用來(lái)記錄鼠標(biāo)左鍵按下的位置
        self.X = tkinter.IntVar(value=0)
        self.Y = tkinter.IntVar(value=0)
        #屏幕尺寸
        screenWidth = root.winfo_screenwidth()
        screenHeight = root.winfo_screenheight()
        #創(chuàng)建頂級(jí)組件容器
        self.top = tkinter.Toplevel(root, width=screenWidth, height=screenHeight)
        #不顯示最大化、最小化按鈕
        self.top.overrideredirect(True)
        self.canvas = tkinter.Canvas(self.top,bg='white', width=screenWidth, height=screenHeight)
        #顯示全屏截圖，在全屏截圖上進(jìn)行區(qū)域截圖
        self.image = tkinter.PhotoImage(file=png)
        self.text =""
        self.canvas.create_image(screenWidth//2, screenHeight//2, image=self.image)
        #鼠標(biāo)左鍵按下的位置
        def onLeftButtonDown(event):
            self.X.set(event.x)
            self.Y.set(event.y)
            #開(kāi)始截圖
            self.sel = True
        self.canvas.bind('Button-1>', onLeftButtonDown)
        #鼠標(biāo)左鍵移動(dòng)，顯示選取的區(qū)域
        def onLeftButtonMove(event):
            if not self.sel:
                return
            global lastDraw
            try:
                #刪除剛畫完的圖形，要不然鼠標(biāo)移動(dòng)的時(shí)候是黑乎乎的一片矩形
                self.canvas.delete(lastDraw)
            except Exception as e:
                pass
            lastDraw = self.canvas.create_rectangle(self.X.get(), self.Y.get(), event.x, event.y, outline='black')
        self.canvas.bind('B1-Motion>', onLeftButtonMove)
        #獲取鼠標(biāo)左鍵抬起的位置，保存區(qū)域截圖
        def onLeftButtonUp(event):
            self.sel = False
            try:
                self.canvas.delete(lastDraw)
            except Exception as e:
                pass
            sleep(0.1)
            #考慮鼠標(biāo)左鍵從右下方按下而從左上方抬起的截圖
            left, right = sorted([self.X.get(), event.x])
            top, bottom = sorted([self.Y.get(), event.y])
            pic = ImageGrab.grab((left+1, top+1, right, bottom))
            fileName ="temp.jpg"
            pic.save(fileName)
            self.text = get_text(fileName)
            #關(guān)閉當(dāng)前窗口
            self.top.destroy()
        self.canvas.bind('ButtonRelease-1>', onLeftButtonUp)
#讓canvas充滿窗口，并隨窗口自動(dòng)適應(yīng)大小
        self.canvas.pack(fill=tkinter.BOTH, expand=tkinter.YES)
 #開(kāi)始截圖
def buttonCaptureClick():
    #最小化主窗口
#     root.state('icon')
    root.withdraw()
    sleep(0.4)
    filename = 'temp.png'
    #grab()方法默認(rèn)對(duì)全屏幕進(jìn)行截圖
    im = ImageGrab.grab()
    im.save(filename)
    im.close()
    #顯示全屏幕截圖
    w = MyCapture(filename)
    root.wait_window(w.top)
    #截圖結(jié)束，恢復(fù)主窗口，并刪除臨時(shí)的全屏幕截圖文件
    root.update()
    root.deiconify()
    text1.config(state = NORMAL)
    text1.delete(0.0,END)
    text1.insert('insert',w.text)
    text1.config(state = DISABLED)
    text1.pack()
    os.remove(filename)

def get_text(img_path):
    print("")
    img = img_path # 圖片路徑
    files = {"pic_path": open(img, "rb")}  # files # 類似data數(shù)據(jù)
    url = "http://pic.sogou.com/pic/upload_pic.jsp"  # post的url
    keywords = requests.post(url, files=files).text  # requests 提交圖片
    url = "http://pic.sogou.com/pic/ocr/ocrOnline.jsp?query=" + keywords  # keywords就是圖片url此方式為get請(qǐng)求
    ocrResult = requests.get(url).json()  # 直接轉(zhuǎn)換為json格式
    
    contents = ocrResult['result']  # 類似字典 把result的value值取出來(lái) 是一個(gè)list然后里面很多json就是識(shí)別的文字
    text = ""
    for content in contents:  # 遍歷所有結(jié)果
        text+=(content['content'].strip()+'\n')  # strip去除空格 他返回的結(jié)果自帶一個(gè)換行
    return text

self.text = get_text(fileName)  

    text1.config(state = NORMAL)
    text1.delete(0.0,END)
    text1.insert('insert',w.text)
    text1.config(state = DISABLED)
    text1.pack()