From 0d5b311d83bf13ece7e6b45051ca217d37bceb44 Mon Sep 17 00:00:00 2001 From: Jiawei Wang Date: Sat, 18 Jul 2020 10:38:37 +0800 Subject: [PATCH] Update ocr_web_client.py --- python/examples/ocr/ocr_web_client.py | 29 --------------------------- 1 file changed, 29 deletions(-) diff --git a/python/examples/ocr/ocr_web_client.py b/python/examples/ocr/ocr_web_client.py index 21bae37d..6ae347a0 100644 --- a/python/examples/ocr/ocr_web_client.py +++ b/python/examples/ocr/ocr_web_client.py @@ -20,47 +20,18 @@ import base64 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( 'utf8') #data.tostring()).decode('utf8') - headers = {"Content-type": "application/json"} url = "http://127.0.0.1:9292/ocr/prediction" 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() -- GitLab