查看: 2303|回复: 0

Python-百度ocr识别

字体大小: 正常放大

檀俾九

7 主题	3 听众	10 积分

升级 5.26%

该用户从未签到

自我介绍: 我本名为我，那就是我

发表于 2021-1-12 09:38 |显示全部楼层

|招呼Ta 关注Ta

#!/usr/bin/python\" ~, ?' S5 g# z; F/ E8 U, _ S
# -*- coding: utf-8 -*-
- O- h4 U8 P7 ] q\" Z/ t5 r/ @\" q. ^8 f& }, Y/ P
import base64
2 Q$ p: w1 ~/ ofrom os.path import exists
9 B: S# w ~3 m* ~4 O, a# Jfrom tkinter import Tk
j5 s4 p: Z$ Afrom tkinter.filedialog import askopenfilename0 I: d( i\" B7 t
from urllib.parse import urlencode
1 `\" _ e* ]8 M0 {: B
! U8 b' ^$ r3 Q; @) _6 n) |import requests
1 I% d: o7 B+ T! t# C
5 e( S$ W5 ~( A$ n5 ]
& v: L( T7 }. z; B& D7 Ddef Dialog_box():
+ S. Y0 ^$ E% I7 X* `/ P# B root = Tk()1 x0 Z: f9 p! @+ b' q
root.withdraw()
4 n {4 W0 {6 S9 _2 Q, Q, L& j+ P% A
filetypes = [(2 c' Z* f E\" ?# E z6 G( S
'图片文件(*.jpg;*.jpeg;*.gif;*.png;*.bmp)', '*.jpg;*.jpeg;*.gif;*.png;*.bmp'),
. \$ p* T- f( P6 q6 j: \ ('所有文件(*.*)', '*.*')]
3 r& ~ j6 F, F) u: N* t% f8 C
. [\" l2 B4 r J$ N file_path = askopenfilename(filetypes=filetypes)
1 ]9 n/ a5 r- ]$ G return file_path
+ L, _+ v/ n; X0 H1 w( Y1 U9 v8 V- k- ~1 _
, t* Q3 h: p9 \
def Read_file(file_path):
+ `$ O8 p u; P- [7 m/ Q, | with open(file_path, mode='rb') as file_object:
) S. A1 H$ m P$ u6 q! j ~ contents = file_object.read()
+ O8 y! m1 L: U! U: s G/ V( C return contents.rstrip()$ s! ]- H, z4 ?
2 A) G) F; Q; c8 c$ o
, }' W1 ^8 M0 b( Z+ J
def Baidu_ocr(binary):
; d! A4 L# x& g' d' f url = "https://aip.baidubce.com/oauth/2.0/token?"
0 R) @2 X+ |( r6 \ data = {3 \& ?- ~' k7 [\" k& ?
"grant_type": "client_credentials", m, ~3 [ |& ^1 D- v
"client_id": "", # 必填
9 E6 ]' M2 I$ p1 `- `9 X "client_secret": "" # 必填
+ L1 Z4 m2 ` O( B7 }+ S. e7 C. k }
: _4 d; u1 x& j {. A) D& L url += urlencode(data)
7 r8 W; M* h# M+ i$ j [0 l response = requests.get(url=url).json()
# F% X- Z- d. z: o4 r x! \ access_token = response['access_token']
5 G1 ~7 c1 x. p; R\" ^\" J% l
( M+ @+ ?; u% U* w- S\" ] url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic?access_token=" + access_token
, ?, g4 s* ]) I$ c7 z! V post_data = {
6 x- l1 {9 Z8 D5 U "image": base64.b64encode(binary),
# b ^; H/ n: |& W "url": "",
; b, Z% y+ [. m* b: n+ { "language_type": "CHN_ENG",
, x4 y1 {2 _\" _8 b\" j "detect_direction": "false",; g6 ?2 Y0 y) L0 G, `
"detect_language": "false",
: I, k5 a( Q1 o5 H0 `+ g "probability": "false". r6 Z9 L2 \+ a8 B' q0 L R$ A4 q2 h
}& U6 u8 s3 [6 A3 g) y& A+ m
response = requests.post(url=url, data=post_data).json()
+ b( c! p0 z4 S( g/ r$ o. A num = response['words_result_num']
: X( A0 g, J5 h; X result = ''6 |5 X4 |1 k$ R- k0 k
for i in range(num):- ~: W. _& _ Z\" t% g$ s( O2 o
result += response['words_result'][ i]['words'] + "\r\n"
\" j) C2 ^/ u6 W return result
. p* U: v4 N. c\" G/ q! a; R
8 N# |! V1 |/ K3 N! U+ k; b
2 b# B2 r3 L% l8 x/ Fif __name__ == "__main__":+ _$ E/ h/ Z+ W7 i& q8 x\" X
file_path = Dialog_box()$ ?8 M+ y7 D1 z/ x& j. U, X$ E
if exists(file_path):
1 _% w, }8 R/ o; U/ S' ]+ x- G data = Read_file(file_path)
: V7 k' y0 f, X8 m2 J print(Baidu_ocr(data))
2 ~& K# m8 y\" _$ Z3 J
( n- O6 d) U! u8 I% W1 _+ ?

#!/usr/bin/python\" ~, ?' S5 g# z; F/ E8 U, _  S 
# -*- coding: utf-8 -*- 
- O- h4 U8 P7 ]  q\" Z/ t5 r/ @\" q. ^8 f& }, Y/ P 
import base64 
2 Q$ p: w1 ~/ ofrom os.path import exists 
9 B: S# w  ~3 m* ~4 O, a# Jfrom tkinter import Tk 
  j5 s4 p: Z$ Afrom tkinter.filedialog import askopenfilename0 I: d( i\" B7 t 
from urllib.parse import urlencode 
1 `\" _  e* ]8 M0 {: B 
! U8 b' ^$ r3 Q; @) _6 n) |import requests 
1 I% d: o7 B+ T! t# C 
5 e( S$ W5 ~( A$ n5 ] 
& v: L( T7 }. z; B& D7 Ddef Dialog_box(): 
+ S. Y0 ^$ E% I7 X* `/ P# B    root = Tk()1 x0 Z: f9 p! @+ b' q 
    root.withdraw() 
4 n  {4 W0 {6 S9 _2 Q, Q, L& j+ P% A 
    filetypes = [(2 c' Z* f  E\" ?# E  z6 G( S 
        '图片文件(*.jpg;*.jpeg;*.gif;*.png;*.bmp)', '*.jpg;*.jpeg;*.gif;*.png;*.bmp'), 
. \$ p* T- f( P6 q6 j: \        ('所有文件(*.*)', '*.*')] 
3 r& ~  j6 F, F) u: N* t% f8 C 
. [\" l2 B4 r  J$ N    file_path = askopenfilename(filetypes=filetypes) 
1 ]9 n/ a5 r- ]$ G    return file_path 
+ L, _+ v/ n; X0 H1 w( Y1 U9 v8 V- k- ~1 _ 
, t* Q3 h: p9 \ 
def Read_file(file_path): 
+ `$ O8 p  u; P- [7 m/ Q, |    with open(file_path, mode='rb') as file_object: 
) S. A1 H$ m  P$ u6 q! j  ~        contents = file_object.read() 
+ O8 y! m1 L: U! U: s  G/ V( C        return contents.rstrip()$ s! ]- H, z4 ? 
2 A) G) F; Q; c8 c$ o 
, }' W1 ^8 M0 b( Z+ J 
def Baidu_ocr(binary): 
; d! A4 L# x& g' d' f    url = "https://aip.baidubce.com/oauth/2.0/token?" 
0 R) @2 X+ |( r6 \    data = {3 \& ?- ~' k7 [\" k& ? 
        "grant_type": "client_credentials",  m, ~3 [  |& ^1 D- v 
        "client_id": "",  # 必填 
9 E6 ]' M2 I$ p1 `- `9 X        "client_secret": ""  # 必填 
+ L1 Z4 m2 `  O( B7 }+ S. e7 C. k    } 
: _4 d; u1 x& j  {. A) D& L    url += urlencode(data) 
7 r8 W; M* h# M+ i$ j  [0 l    response = requests.get(url=url).json() 
# F% X- Z- d. z: o4 r  x! \    access_token = response['access_token'] 
5 G1 ~7 c1 x. p; R\" ^\" J% l 
( M+ @+ ?; u% U* w- S\" ]    url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic?access_token=" + access_token 
, ?, g4 s* ]) I$ c7 z! V    post_data = { 
6 x- l1 {9 Z8 D5 U        "image": base64.b64encode(binary), 
# b  ^; H/ n: |& W        "url": "", 
; b, Z% y+ [. m* b: n+ {        "language_type": "CHN_ENG", 
, x4 y1 {2 _\" _8 b\" j        "detect_direction": "false",; g6 ?2 Y0 y) L0 G, ` 
        "detect_language": "false", 
: I, k5 a( Q1 o5 H0 `+ g        "probability": "false". r6 Z9 L2 \+ a8 B' q0 L  R$ A4 q2 h 
    }& U6 u8 s3 [6 A3 g) y& A+ m 
    response = requests.post(url=url, data=post_data).json() 
+ b( c! p0 z4 S( g/ r$ o. A    num = response['words_result_num'] 
: X( A0 g, J5 h; X    result = ''6 |5 X4 |1 k$ R- k0 k 
    for i in range(num):- ~: W. _& _  Z\" t% g$ s( O2 o 
        result += response['words_result'][ i]['words'] + "\r\n" 
\" j) C2 ^/ u6 W    return result 
. p* U: v4 N. c\" G/ q! a; R 
8 N# |! V1 |/ K3 N! U+ k; b 
2 b# B2 r3 L% l8 x/ Fif __name__ == "__main__":+ _$ E/ h/ Z+ W7 i& q8 x\" X 
    file_path = Dialog_box()$ ?8 M+ y7 D1 z/ x& j. U, X$ E 
    if exists(file_path): 
1 _% w, }8 R/ o; U/ S' ]+ x- G        data = Read_file(file_path) 
: V7 k' y0 f, X8 m2 J        print(Baidu_ocr(data)) 
2 ~& K# m8 y\" _$ Z3 J 
( n- O6 d) U! u8 I% W1 _+ ?

转发自派生社区

Python交流群：1047602540

zan

使用道具举报