diff --git a/python/examples/ocr/ocr_web_client.py b/python/examples/ocr/ocr_web_client.py index a8f382b29b5f6d962a911f6194f53026ed420075..21bae37d7d1ff5600db090029b9d353eb187f7f5 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 Binary files /dev/null and b/python/examples/ocr/rec_img/ch_doc3.jpg differ