![]()
import requests
from PIL import Image
import json
import base64
import numpy as np
import cv2
import re
import os
import glob
from tqdm import tqdm
import math
import time
from gradio.processing_utils import encode_pil_to_base64
def decode_image(src):
??"""
??解碼圖片
??:param src: 圖片編碼
??:return: str 保存到本地的文件名
??"""
??# 1、信息提取
??result = re.search("data:image/(?P.*?);base64,(?P.*)", src, re.DOTALL)
??if result:
????ext = result.groupdict().get("ext")
????data = result.groupdict().get("data")
??else:
????raise Exception("Do not parse!")
??# 2、base64解碼
??img = base64.urlsafe_b64decode(data)
??return img
??# 3、二進(jìn)制文件保存
??# with open(save_path, "wb") as f:
??#???f.write(img)
def text2img(url, prompt):
??payload = {
??"enable_hr": False,
??"denoising_strength": 0,
??"firstphase_width": 0,
??"firstphase_height": 0,
??"hr_scale": 2,
??"hr_upscaler": "string",
??"hr_second_pass_steps": 0,
??"hr_resize_x": 0,
??"hr_resize_y": 0,
??"prompt": prompt,
??"styles": [
????""
??],
??"seed": -1,
??"subseed": -1,
??"subseed_strength": 0,
??"seed_resize_from_h": -1,
??"seed_resize_from_w": -1,
??"batch_size": 1,
??"n_iter": 1,
??"steps": 50,
??"cfg_scale": 7,
??"width": 512,
??"height": 512,
??"restore_faces": False,
??"tiling": False,
??"negative_prompt": "nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry",
??"eta": 0,
??"s_churn": 0,
??"s_tmax": 0,
??"s_tmin": 0,
??"s_noise": 1,
??"override_settings": {},
??"override_settings_restore_afterwards": True,
??"script_args": [],
??"sampler_index": "Euler",
??}
??req = requests.post(url = url, json = payload)
??return json.loads(req.text)
def text2img_example():
??# api 請(qǐng)求地址，server 啟動(dòng)命令 python launch.py --api 增加一個(gè)api參數(shù)
??txt2img_url = "http://127.0.0.1:7860/sdapi/v1/txt2img"
??now = time.time()
??prompt = "1girl, beautiful, car, simple background"
??res = text2img(txt2img_url, prompt)
??end = time.time()
??print(end-now)
??gn_img_fname = "./"
??# print(res)
??for idx, image in enumerate(res['images']):
????# print(image)
????img = base64.urlsafe_b64decode(image)
????# gn_img = decode_image(image)
????# print(gn_img)
????gn_img_save_path = str(idx) + ".jpg"
????with open(gn_img_save_path, "wb") as f:
??????f.write(img)
if __name__ == "__main__":
??text2img_example()
標(biāo)簽：
