forked from BIGPPWONG/idcardocr
-
Notifications
You must be signed in to change notification settings - Fork 0
/
idcard_recognize.py
86 lines (76 loc) · 2.72 KB
/
idcard_recognize.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# -*- coding: utf-8 -*-
import idcardocr
import findidcard
import json
from http.server import BaseHTTPRequestHandler, HTTPServer
import socketserver
import cv2, time
import uuid
import cgi
def process(img_name):
try:
idfind = findidcard.findidcard()
idcard_img = idfind.find(img_name)
result_dict = idcardocr.idcardocr(idcard_img)
result_dict['error'] = 0
except Exception as e:
result_dict = {'error':1}
print(e)
return result_dict
#SocketServer.ForkingMixIn, SocketServer.ThreadingMixIn
class ForkingServer(socketserver.ForkingMixIn, HTTPServer):
pass
class S(BaseHTTPRequestHandler):
def _set_headers(self):
self.send_response(200)
self.send_header('Content-type', 'application/json')
#self.end_headers()
def do_GET(self):
self._set_headers()
# self.wfile.write("<html><body><h1>hi!</h1></body></html>")
def do_HEAD(self):
self._set_headers()
def do_POST(self):
#content_length = int(self.headers['Content-Length']) # <--- Gets the size of data
# post_data = self.rfile.read(content_length) # <--- Gets the data itself
ctype, pdict = cgi.parse_header(self.headers['content-type'])
print(pdict)
pdict['boundary'] = bytes(pdict['boundary'], "utf-8")
multipart_data = cgi.parse_multipart(self.rfile, pdict)
filename = uuid.uuid1()
fo = open("tmp/%s.jpg"%filename, "wb")
# print(str(multipart_data))
# print(multipart_data.get('pic')[0])
fo.write( multipart_data.get('pic')[0] )
fo.close()
result = process("tmp/%s.jpg"%filename)
#print result
self._set_headers()
self.send_header("Content-Length", str(len(json.dumps(result).encode('utf-8'))))
self.end_headers()
self.wfile.write(json.dumps(result).encode('utf-8'))
def http_server(server_class=ForkingServer, handler_class=S, port=8080):
server_address = ('', port)
httpd = server_class(server_address, handler_class)
cv2.ocl.setUseOpenCL(False)
print('Starting httpd...')
print(u"是否启用OpenCL：%s"%cv2.ocl.useOpenCL())
httpd.serve_forever()
if __name__=="__main__":
# p = Pool()
#r9 = p.apply_async(process, args=('9.jpg',))
#r14 = p.apply_async(process, args=('14.jpg',))
# p.apply_async(http_server)
# p.apply_async(http_server)
# p.apply_async(http_server)
# p.apply_async(http_server)
# p.close()
# p.join()
#print r9.get(), r14.get()
http_server()
# cv2.ocl.setUseOpenCL(True)
# t1 = round(time.time() * 1000)
# for i in range(1,15):
# print(process('./testimages/%s.jpg'%i))
# t2 = round(time.time() * 1000)
# print('time:%s' % (t2 - t1))