代碼：

import base64
import datetime
import json
import os

import requests


def submit_post(url: str, data: dict):
    """
    Submit a POST request to the given URL with the given data.
    :param url:  url
    :param data: data
    :return:  response
    """
    return requests.post(url, data=json.dumps(data))


def save_encoded_image(b64_image: str, output_path: str):
    """
    Save the given image to the given output path.
    :param b64_image:  base64 encoded image
    :param output_path:  output path
    :return:  None
    """
    # 判斷當(dāng)前目錄下是否存在 output 文件夾，如果不存在則創(chuàng)建
    if not os.path.exists("output"):
        os.mkdir("output")
    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
    output_path = f"{output_path}_{timestamp}" + ".png"
    # 將文件放入當(dāng)前目錄下的 output 文件夾中
    output_path = f"output/{output_path}"
    with open(output_path, "wb") as f:
        f.write(base64.b64decode(b64_image))


def save_json_file(data: dict, output_path: str):
    """
    Save the given data to the given output path.
    :param data:  data
    :param output_path:  output path
    :return:  None
    """
    # 忽略 data 中的 images 字段
    data.pop('images')
    # 將 data 中的 info 字段轉(zhuǎn)為 json 字符串，info 當(dāng)前數(shù)據(jù)需要轉(zhuǎn)義
    data['info'] = json.loads(data['info'])

    # 輸出 data.info.infotexts
    info_texts = data['info']['infotexts']
    for info_text in info_texts:
        print(info_text)

    # 判斷當(dāng)前目錄下是否存在 output 文件夾，如果不存在則創(chuàng)建
    if not os.path.exists("output"):
        os.mkdir("output")
    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
    output_path = f"{output_path}_{timestamp}" + ".json"
    # 將文件放入當(dāng)前目錄下的 output 文件夾中
    output_path = f"output/{output_path}"
    with open(output_path, "w") as f:
        json.dump(data, f, indent=4, ensure_ascii=False)


if __name__ == '__main__':
    """
    Example usage: python3 txt2img.py
    """
    txt2img_url = "http://192.168.2.82:7860/sdapi/v1/txt2img" # 服務(wù)器地址
    prompt = input("請(qǐng)輸入提示詞：")
    negative_prompt = input("請(qǐng)輸入反面提示詞：")
    data = {'prompt': prompt, 'negative_prompt': negative_prompt}
    # 將 data.prompt 中的文本，刪除文件名非法字符，已下劃線分隔，作為文件名
    output_path = data['prompt'].replace(" ", "_").replace("/", "_").replace("\\", "_").replace(":", "_").replace("\"",
                                                                                                                  "_").replace(
        "<", "_").replace(">", "_").replace("|", "_")
    response = submit_post(txt2img_url, data)
    save_encoded_image(response.json()['images'][0], output_path)
    save_json_file(response.json(), output_path)

輸出：

>> python txt2img.py
請(qǐng)輸入提示詞：masterpiece, best quality, 1boy, orc, male focus, mecha, robot, science fiction, solo, black background <lora:nijiMecha:1> <lora:wzh-000022:1>
請(qǐng)輸入反面提示詞：(worst quality, low quality:1.4), greyscale, monochrome, watermark, signature

masterpiece, best quality, 1boy, orc, male focus, mecha, robot, science fiction, solo, black background <lora:nijiMecha:1> <lora:wzh-000022:1>
Negative prompt: (worst quality, low quality:1.4), greyscale, monochrome, watermark, signature
Steps: 50, Sampler: Euler, CFG scale: 7.0, Seed: 274772779, Size: 512x512, Model hash: ce67b54cf2, Model: oukaNiji5_v10, Seed resize from: -1x-1, Denoising strength: 0, Clip skip: 2

說明：

• 運(yùn)行后，圖片以及 JSON 將會(huì)輸出到當(dāng)前目錄下 output 中；

TIP：文章來源地址http://www.zghlxwxcb.cn/news/detail-568059.html

• 當(dāng)然前提是你已經(jīng)部署好 Stable Diffusion API 服務(wù)；
• 并且安裝好跟我一樣的模型以及一些相關(guān)的 LoRA；

print('HELLO WORLD') # http://www.skyner.cn

到了這里，關(guān)于Python 使用 Stable Diffusion API 生成圖片示例的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！