AI为图片添加标签HTTP调用
带模型压缩后600MB不到，只适用于动漫图片标签标记。整合包：https://pan.baidu.com/s/1MOQJiVtMdYKfjWUiqmI3JA?pwd=moxi
抢个板凳，感谢楼主分享，收藏了先！！ 这个好 可以整理下硬盘里的图了 收藏了，感谢分享 无限开cmd窗口，弹死我了
感谢分享试试看 yhlg 发表于 2022-12-25 09:49
无限开cmd窗口，弹死我了
编译的时候可能写了个-w，你重新编译看看
# coding=utf-8
from deepdanbooru_onnx import DeepDanbooru
from PIL import Image
import json,io,os,sys,time,re
import http.server
import configparser
import threading
import socket
occupancy = 27513
port = 13609
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
def check_port(oport):# bianyi -F ImageTagger.py
try:# bianyi ImageTagger.py
s.bind(('localhost', oport))
return True
except OSError:
return False
if not check_port(occupancy):
print(f'重复运行')
#time.sleep(5)
sys.exit()
print('DeepDanbooru...Init')
config = configparser.ConfigParser()
if os.path.exists('DeepDanbooru.ini'):
config.read('DeepDanbooru.ini')
else:
config['Proxy'] = {'Proxy':'127.0.0.1:1080'}
with open('DeepDanbooru.ini','w') as configfile:
config.write(configfile)
config.read('DeepDanbooru.ini')
Proxy = config.get('Proxy','Proxy')
try:
root = os.getcwd()
if not os.path.exists(root+'/model'):
os.mkdir(root+'/model')
model_path=root+'/model/deepdanbooru.onnx'
tags_path=root+'/model/tags.txt'
if not os.path.exists(model_path):
model_path = None
if not os.path.exists(tags_path):
tags_path = None
danbooru = DeepDanbooru(Proxy=Proxy,model_path=model_path,tags_path=tags_path)
except Exception as e:
print(':'+Proxy)
print(":",e)
time.sleep(100)
sys.exit()
print('DeepDanbooru...End')
class ImageServer(http.server.BaseHTTPRequestHandler):
def do_POST(self):
# 读取请求体中的数据
if self.headers['Content']=="End":#关闭
t = threading.Thread(target=shutdown_server)
s.close()
t.start()
sys.exit()
length = int(self.headers['Content-Length'])
data = self.rfile.read(length)
if data:
img = Image.open(io.BytesIO(data))
res = danbooru(img)
self.send_response(200)
self.send_header('Content-Type', 'text/json')
self.end_headers()
res=sortJsonData(res)
otl,tlv = {},{}
for item in res:
tlv=transl
otl['original'] = res
otl['control'] = tlv
result = re.match('^"([\s\S]*?)"$', json.dumps(str(otl).replace("'", '"')))
self.wfile.write(bytes(result.group(1),'UTF-8'))
else:
self.send_response(200)
self.send_header('Content-Type', 'text/plain')
self.end_headers()
self.wfile.write(bytes('Me is Online','UTF-8'))
# 发送响应
# 创建并启动服务器
def shutdown_server():
server.shutdown()
global transl
transl = None
if os.path.exists('./transl.json'):
with open('./transl.json', 'r', encoding='utf-8') as f:
transl = json.load(f)
def sortJsonData(jsondata):
sorted_jsondata = sorted(jsondata.items(), key=lambda item:item, reverse=True)
return dict(map(lambda x: (x, x), sorted_jsondata))
server = http.server.HTTPServer(('localhost', port), ImageServer)
print('Server:127.0.0.1:'+str(port))
server.serve_forever()
页:
[1]