Skip to content

C
cog-anytext
Commit fa8f12c2 authored by duanjinfei's avatar duanjinfei
Browse files
Options

update predict

parent 3e253ae6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
util.py
View file @ fa8f12c2
...@@ -45,10 +45,11 @@ def save_images(img_list, folder): ...@@ -45,10 +45,11 @@ def save_images(img_list, folder):
def read_img_result(img_list): def read_img_result(img_list):
res = [] res = []
for _, img in enumerate(img_list): for _, img in enumerate(img_list):
_, img_encoded = cv2.imencode('.jpg', img) _, img_encoded = cv2.imencode('.jpg', img[..., ::-1])
# Encode image as base64 # Encode image as base64
base64_encoded = base64.b64encode(img_encoded).decode('utf-8') base64_encoded = base64.b64encode(img_encoded).decode('utf-8')
res.append(base64_encoded) mime_prefix = 'data:image/jpg;base64,'
res.append(mime_prefix + base64_encoded)
return res return res
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment