pipeline_http_client.py 501 字节
Newer Older
1 2 3 4 5
import requests
import json
import base64
import os

S
stephon 已提交
6

7 8 9
def cv2_to_base64(image):
    return base64.b64encode(image).decode('utf8')

S
stephon 已提交
10

11 12 13 14 15 16
if __name__ == "__main__":
    url = "http://127.0.0.1:18080/imagenet/prediction"
    with open(os.path.join(".", "daisy.jpg"), 'rb') as file:
        image_data1 = file.read()
    image = cv2_to_base64(image_data1)
    data = {"key": ["image"], "value": [image]}
S
stephon 已提交
17
    for i in range(1):
18 19
        r = requests.post(url=url, data=json.dumps(data))
        print(r.json())