From bc6488d71600cc071457a22de4488ab75a6dc6f6 Mon Sep 17 00:00:00 2001 From: dyning Date: Sat, 18 Jul 2020 10:30:40 +0800 Subject: [PATCH] add ocr/rec test cases --- python/examples/ocr/ocr_web_client.py | 30 ++++++++++++++++++++++-- python/examples/ocr/rec_img/ch_doc3.jpg | Bin 0 -> 2117 bytes 2 files changed, 28 insertions(+), 2 deletions(-) create mode 100644 python/examples/ocr/rec_img/ch_doc3.jpg diff --git a/python/examples/ocr/ocr_web_client.py b/python/examples/ocr/ocr_web_client.py index a8f382b2..21bae37d 100644 --- a/python/examples/ocr/ocr_web_client.py +++ b/python/examples/ocr/ocr_web_client.py @@ -21,6 +21,25 @@ import os, sys import time +class Timetool(object): + def __init__(self, filename): + self.filename = filename + self.f = open(filename, "w+") + self.buf = "" + + def write(self): + self.f.write(self.buf) + self.f.close() + self.buf = "" + self.f = open(self.filename, "a") + + def record(self, stage): + self.buf += "%s : %.6f\n" % (stage, time.time()) + + def next(self): + self.buf += "-----\n" + + def cv2_to_base64(image): #data = cv2.imencode('.jpg', image)[1] return base64.b64encode(image).decode( @@ -29,12 +48,19 @@ def cv2_to_base64(image): headers = {"Content-type": "application/json"} url = "http://127.0.0.1:9292/ocr/prediction" -test_img_dir = "test_imgs/rctw_samples/" - +test_img_dir = "imgs/" +t = Timetool("cli_time.log") +count = 0 for img_file in os.listdir(test_img_dir): + count += 1 + t.record("client start") with open(os.path.join(test_img_dir, img_file), 'rb') as file: image_data1 = file.read() image = cv2_to_base64(image_data1) + t.record("client image read") data = {"feed": [{"image": image}], "fetch": ["res"]} r = requests.post(url=url, headers=headers, data=json.dumps(data)) print(r.json()) + t.record("client response") +if count == 500: + t.write() diff --git a/python/examples/ocr/rec_img/ch_doc3.jpg b/python/examples/ocr/rec_img/ch_doc3.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c0c2053643c6211b9c2017e305c5fa05bba0cc66 GIT binary patch literal 2117 zcmbW$c{tSj9tZH>Y-Y@otWDD(OW8ukFqA2d8Z$^)GRN35gG5gDEviYDh6<%F!bq|m zX=26{Eo!J-W9QhWlWjz5IEPBQxw`j0=RVK<>zw=kKHtASpXd90UeEXWtPiih0X92Q zXcPbh0s+T14zNB3*aHYTIk+4Q0f)oofrp|>h2E2yaBG&R(5c>H!<6TR)) zyL9k)eUibh-DX50QB%*FY(=m!F((qfKLV1Mmq(+}SPTYB(86mG{&Loz0*VM=I|vH_ zX#-$I5JVBQE&? z2M$sks5D1+kE5RSAH5i?z@XscAtyqkV`Agt6B3hBFQlbsWM1UtUA>ln{l?!43U8I) zuDDZKRb6wxp^-0W5P=b;Q~XU5Gee+3j~hexDZ7sOv_A8$<7rX5UH$9U?Wt1$Svcw z%IlcBEvho3dXZ|n#EBhC-)aAn{W;kA|4a4%v**bfYgDgBh}ny#`U zu3nw;%bWv!)=$C0wSM9kX=%(t$0aRf$pC0X%CoVX)F(Qnh!XQY+6@WT=%opHADoPx z*`b$%_IlDb*>W+LG~2I_ik4cTN{JVeSMkQNIu(hQY=HA+>GkTd=Yb0{zx)dSvDc$l z8pP6tH-*3xO3EBosZu@CFLqLSl1z$ua#9qvPtlnxvpI;V)za7ZRu0A4 z)!lw~>fy-#coa$UQf8}hMY(`UZ{3z2abl%AgdX42-+15Wz!IpVGPG}yZ?#-xZ9?(9k1E{dSZJX=JPriAW0VdAnSa_3xf&jIH6uSObti|jmcJ)__!4GRz0 z-%02YNbiXgD#_f`Tuynd|D}CxGoxO-G$57;(rBo~7)o{+Jc$km*FV=QLQ*Lt^khqe zh=f~eWZ!*Coz5XBk2arDIJ8U6(NxCF9ouFb&<4b)7MYamx~l|f8qCfyQlWuSMx>*H z@sg;^blGVR&_mh8X20RY8JW_A9a!viyL z%V@5C@l`&)^ZW3^!f&f?2TvwypWFKnQ}@f?+6_kXx3pzMUo%0yQ8eQs^Qu|~P z8uGNB+Nbp()j7?iG`aN*%XUi*@9{Ysj*o(=?DWxa7Cp;>|GU)lmrdYj~uG>|`w^IgWJaD^rFG`;KzE#M7q8D$_A-;P$@{%}?W z6c|%Om>2N;D~cZPjLcVMeDYe`S&g}o=iRwgGECoJ=9CMybaQ?oob%~=bZQb4 zN_AxDMz)U$t`0TVm0Q0cK6(qu31^Aaj@&9X^U;UQSoDoLf4YwRP3xj0Q6v8p1kZ71 zak-lqzDh&Jabur?vI3hmR{YyWqZ-vByvJ&R)sY_Df~8hO8)c$raPOD`|1gxzo54>F zo!h-^SIv;<>K*2w+^h|T>Jzfu)?nAw6f7OAj!-CJArWS-uh7wGDmz2SRj}|T?eYD3 zWAB-s1%mtMP$MHxr)Gaa`WpVYb};^`%=b%(TINnb>Hf`o7R*^1zolV~$Vsqiep%f%~` zi%DQ@e#SdbbI5~y zT~=XMyt75_rXcRJ{1+t`1Y6t@4uxN(4{xH6k{#tub_H=vA3x9Tkqgc!A8x%fCTu!v zMwhV)$Gfp@9{m$}DD17PvZ~fL1{dwG`0FS_HIy%gr;>^yx`~qUk`OQ|+`)yngicDyf2gd#TD;el!8n_K TlDIY%k1RFEPD?q!`p7>4iY2E| literal 0 HcmV?d00001 -- GitLab