diff --git a/qwerqwe.jpg b/qwerqwe.jpg new file mode 100644 index 0000000..f01fe02 Binary files /dev/null and b/qwerqwe.jpg differ diff --git a/test_color.py b/test_color.py index 80b6a7f..2b34922 100644 --- a/test_color.py +++ b/test_color.py @@ -7,7 +7,8 @@ import random # url = 'http://172.17.110.92/hairColor/v2' -url = 'https://693565557084165-http-8801.northwest1.gpugeek.com:8443/hairColor/v2' +#url = 'https://693565557084165-http-8801.northwest1.gpugeek.com:8443/hairColor/v2' +url = 'http://localhost:8801/hairColor/v2' @@ -51,7 +52,7 @@ def image_to_base64(file_path, mime_type=None): # 组合成Data URI格式 return f"data:{mime_type};base64,{encoded_string}" -img64_str = image_to_base64("aaa.jpg") +img64_str = image_to_base64("qwerqwe.jpg") data['img'] = img64_str # 存储所有请求的响应 @@ -115,10 +116,31 @@ avgTime = totalReqTime/len(responses_time) print(f"平均相应时间:{avgTime}") +def save_base64_image(base64_str, filename): + """ + 将 Base64 编码的图片字符串保存为本地 JPG 文件。 + + :param base64_str: Base64 编码的图片字符串(可能包含 data:image/jpeg;base64, 前缀) + :param filename: 保存的文件名(例如 'image.jpg') + """ + # 如果包含 data URL 前缀,去掉它 + if base64_str.startswith("data:image"): + base64_str = base64_str.split(",", 1)[1] + + try: + image_data = base64.b64decode(base64_str) + with open(filename, "wb") as f: + f.write(image_data) + print(f"图片已保存为 {filename}") + except Exception as e: + print(f"保存图片失败: {e}") # 打印所有响应 for i, resp in enumerate(responses, 1): print(f"请求 {i} 结果:", {resp.status_code, resp.text[:128]}) + json_obj = json.loads(resp.text) + data_string = json_obj['result'] + save_base64_image(data_string, f"color_result{i}.jpg") print(f"num_requests:{num_requests} total_time:{total_time} qps:{qps}") print(f"平均相应时间:{avgTime}") \ No newline at end of file diff --git a/test_hair.py b/test_hair.py index bcec0a9..8490489 100644 --- a/test_hair.py +++ b/test_hair.py @@ -5,14 +5,15 @@ import base64 import time import random -# url = 'http://mq.aidigifi.meidaojia.com/api/swapHair/v1' -url = 'https://693570664882181-http-8801.northwest1.gpugeek.com:8443/api/swapHair/v1' +url = 'http://localhost:8801/api/swapHair/v1' +# url = 'https://693570664882181-http-8801.northwest1.gpugeek.com:8443/api/swapHair/v1' # url = 'https://692502023221253-http-8801.northwest1.gpugeek.com:8443/api/swapHair/v1' # url = 'https://692493464571909-http-8801.northwest1.gpugeek.com:8443/api/swapHair/v1' headers = {'Content-Type': 'application/json'} data = { # "hair_id": "1907651680352395265", - "hair_id": "1907651525024735234", + # "hair_id": "1907651525024735234", + "hair_id": "1907651680352395265", "task_id": "345564352345", #"user_img_path": "https://cdn.meidaojia.com/ZoeFiles/user2_1_%E5%89%AF%E6%9C%AC.JPG", # "user_img_path": "http://git.xiangsilian.com/imgs/test.jpg", @@ -110,7 +111,9 @@ def image_to_base64(file_path, mime_type=None): # 组合成Data URI格式 return f"data:{mime_type};base64,{encoded_string}" -img64_str = image_to_base64("testuser_8.JPG") +# img64_str = image_to_base64("testuser_8.JPG") +# img64_str = image_to_base64("aaa.jpg") +img64_str = image_to_base64("qwerqwe.jpg") data['user_img_path'] = img64_str # 存储所有请求的响应