iis服务器助手广告广告
返回顶部
首页 > 资讯 > 后端开发 > Python >Stable Diffusion WebUI使用python调用WD 1.4标签器(tagger),获取并处理标签tag权重
  • 947
分享到

Stable Diffusion WebUI使用python调用WD 1.4标签器(tagger),获取并处理标签tag权重

stablediffusionpython数学建模 2023-09-01 11:09:12 947人浏览 薄情痞子

Python 官方文档:入门教程 => 点击学习

摘要

Stable Diffusion为秋叶大佬的绘世2.2.4 WEBUI api后台:Http://127.0.0.1:7860/docs 数据获取结果,python代码在文章尾部 1girl:

Stable Diffusion为秋叶大佬的绘世2.2.4
WEBUI api后台:Http://127.0.0.1:7860/docs

数据获取结果,python代码在文章尾部

1girl: 0.9883618950843811, 98%solo: 0.9468605518341064, 94%horns: 0.9203381538391113, 92%braid: 0.7536494731903076, 75%brown_hair: 0.7361204624176025, 73%sensitive: 0.7181869745254517, 71%looking_at_viewer: 0.6558270454406738, 65%long_hair: 0.6555134654045105, 65%portrait: 0.5619801878929138, 56%hair_ornament: 0.5276427268981934, 52%lips: 0.5271897912025452, 52%realistic: 0.47530364990234375, 47%brown_eyes: 0.44382530450820923, 44%fur_trim: 0.44058263301849365, 44%red_hair: 0.4004508852958679, 40%upper_body: 0.39194822311401367, 39%mole: 0.35748565196990967, 35%general: 0.2813188433647156, 28%questionable: 0.004140794277191162, 0%explicit: 0.0005668997764587402, 0%

使用/tagger/v1/interrogate,先使用get方法获取model模组有十多个,然后把JSON_data提交上去就可以了。记得把图片转码为base64。本文章仅用于测试,请仔细阅读api docs,model和threshold按照需求调整即可

在这里插入图片描述
在这里插入图片描述

import requestsimport base64from collections import OrderedDictfrom PIL import Imageurl = 'http://127.0.0.1:7860/tagger/v1/interrogate'image_path = 'D:/code/image/6.jpg'model = 'wd14-convnext'threshold = 0.35#确认照片为上传照片image = Image.open(image_path)image.show()# 将图片转换为Base64字符串with open(image_path, 'rb') as file:    image_data = file.read()    base64_image = base64.b64encode(image_data).decode('utf-8')# 构建请求体的jsON数据data = {    "image": base64_image,    "model": model,    "threshold": threshold}# 发送POST请求response = requests.post(url, json=data)# 检查响应状态码if response.status_code == 200:    json_data = response.json()    # 处理返回的JSON数据    caption_dict = json_data['caption']    sorted_items = sorted(caption_dict.items(), key=lambda x: x[1], reverse=True)    output = '\n'.join([f'{k}: {v}, {int(v * 100)}%' for k, v in sorted_items])    print(output) else:    print('Error:', response.status_code)    print('Response body:', response.text)

来源地址:https://blog.csdn.net/weixin_42375356/article/details/130904119

--结束END--

本文标题: Stable Diffusion WebUI使用python调用WD 1.4标签器(tagger),获取并处理标签tag权重

本文链接: https://www.lsjlt.com/news/388199.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作