AI开发平台MODELARTS-SDXL基于DevServer适配MindSpore-Lite NPU推理指导（6.3.T041）:Step6 使用Python脚本进行测试

时间：2024-07-06 17:52:04

AI开发平台MODELARTS

Step6 使用Python脚本进行测试

使用python脚本进行测试，脚本内容如下。

# coding=utf-8
import requests
import base64
import os
from io import BytesIO
from PIL import Image


def base64_to_image(base64_str):
    image = base64.b64decode(base64_str, altchars=None, validate=False)
    image = BytesIO(image)
    image = Image.open(image)
    image.save(os.path.join("./outputs","output_img.png"))

if __name__ == '__main__':
    # Config url, token and file path
    url = "http://localhost:8446"

    # Set body,then send request
    headers = {
        'Content-Type':'application/json'
    }
    body = {
     "prompt": "a tiger sleep under the tree",
     "height": 512,
     "width": 512
    }

    resp = requests.post(url, headers=headers, json=body)

    # Print result
    #print(resp.content)
    base64_to_image(resp.content)

将该脚本内容编辑成test.py文件并运行，服务端会打印推理时间，并在outputs文件夹下生成对应图片，打开图片查看效果。