add ocr/rec test cases

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()