add pngs
This commit is contained in:
bridge
@@ -1,5 +1,6 @@
|
||||
import os
|
||||
import base64
|
||||
import random
|
||||
from datetime import datetime
|
||||
|
||||
import requests
|
||||
@@ -28,6 +29,9 @@ def generate_qwen_image(prompt: str, *, size: str = "1328*1328") -> str:
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {API_KEY}",
|
||||
}
|
||||
|
||||
seed = random.randint(1, 4294967290)
|
||||
|
||||
payload = {
|
||||
"model": MODEL,
|
||||
"input": {
|
||||
@@ -45,6 +49,7 @@ def generate_qwen_image(prompt: str, *, size: str = "1328*1328") -> str:
|
||||
"prompt_extend": True,
|
||||
"watermark": True,
|
||||
"size": size,
|
||||
"seed": seed,
|
||||
},
|
||||
}
|
||||
|
||||
@@ -115,43 +120,111 @@ def save_generated_image(query: str, folder: str = "tools/img_gen/tmp/raw") -> s
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
female_prompt_base = "一个好看的仙侠女性头像。只有头部和面部。二次元风格的漫画图片，略微Q版，正面看镜头。纯白背景。像素风格，细节别太多。"
|
||||
female_prompt_base = "一个好看的仙侠女性头像。只有头部和面部且完整露出头部。二次元风格的漫画图片，略微Q版，正面看镜头。纯白背景。像素风格，细节别太多。"
|
||||
# female_affixes = [
|
||||
# "紫色长发，表情嗔怒，带有一丝冷峻，有一个簪子。",
|
||||
# "乌黑直发，眉心一点红砂，清冷淡漠，镶玉步摇。",
|
||||
# "银白短发，英气微笑，发梢轻卷，耳坠为小灵铃。",
|
||||
# "墨绿长发，高马尾，目光坚毅，额前碎发，佩青竹簪。",
|
||||
# "渐变粉蓝长卷发，眸有星点，温柔含笑，薄纱额饰。",
|
||||
# "赤红披发，英气冷艳，眉尾上挑，凤羽发冠。",
|
||||
# "浅金长发，缎带系发，气质圣洁，流苏步摇。",
|
||||
# "乌青长发，微皱眉，眼尾红妆，一枚冰晶发卡。",
|
||||
# "白发如雪，神情淡然，眉心月印，玉质头箍。",
|
||||
# "靛蓝长发，俏皮眨眼，脸颊淡粉，葫芦小发簪。",
|
||||
# "茶棕双丸子头，活泼微笑，脸上淡淡雀斑，小葵花发卡。",
|
||||
# "青丝长发半披半挽，清雅端庄，蝶形玉簪。",
|
||||
# "淡紫短波浪发，俏皮吐舌，星月耳饰。",
|
||||
# "墨发低侧马尾，冷静专注，细链额饰垂坠。",
|
||||
# "湖绿挑染长发，狡黠微笑，狐耳发饰点缀。",
|
||||
# "灰蓝长直发，平刘海，面无表情，银环头饰。",
|
||||
# ]
|
||||
female_affixes = [
|
||||
"紫色长发，表情嗔怒，带有一丝冷峻，有一个簪子。",
|
||||
"乌黑直发，眉心一点红砂，清冷淡漠，镶玉步摇。",
|
||||
"银白短发，英气微笑，发梢轻卷，耳坠为小灵铃。",
|
||||
"墨绿长发，高马尾，目光坚毅，额前碎发，佩青竹簪。",
|
||||
"渐变粉蓝长卷发，眸有星点，温柔含笑，薄纱额饰。",
|
||||
"赤红披发，英气冷艳，眉尾上挑，凤羽发冠。",
|
||||
"浅金长发，缎带系发，气质圣洁，流苏步摇。",
|
||||
"乌青长发，微皱眉，眼尾红妆，一枚冰晶发卡。",
|
||||
"白发如雪，神情淡然，眉心月印，玉质头箍。",
|
||||
"靛蓝长发，俏皮眨眼，脸颊淡粉，葫芦小发簪。",
|
||||
"茶棕双丸子头，活泼微笑，脸上淡淡雀斑，小葵花发卡。",
|
||||
"青丝长发半披半挽，清雅端庄，蝶形玉簪。",
|
||||
"淡紫短波浪发，俏皮吐舌，星月耳饰。",
|
||||
"墨发低侧马尾，冷静专注，细链额饰垂坠。",
|
||||
"湖绿挑染长发，狡黠微笑，狐耳发饰点缀。",
|
||||
"灰蓝长直发，平刘海，面无表情，银环头饰。",
|
||||
"墨黑长发，眼神清澈，嘴角微扬，佩戴白花发饰。",
|
||||
"银灰卷发，神情慵懒，眼角泪痣，水晶耳坠。",
|
||||
"酒红短发，英气勃发，剑眉入鬓，金色额饰。",
|
||||
"浅紫双马尾，活泼可爱，笑眼弯弯，铃铛发带。",
|
||||
"青色长直发，面容清冷，毫无表情，玉簪挽发。",
|
||||
"金棕波浪发，温柔婉约，眉目含情，珍珠步摇。",
|
||||
"深蓝盘发，端庄典雅，气质高贵，凤凰发冠。",
|
||||
"纯白长发，双瞳异色，神秘莫测，银链额饰。",
|
||||
"粉色丸子头，天真烂漫，脸颊红晕，桃花发卡。",
|
||||
"亚麻色碎发，眼神坚毅，嘴角紧抿，简约发带。",
|
||||
"橙红长发，热情似火，笑容灿烂，火焰纹饰。",
|
||||
"墨绿麻花辫，恬静自然，怀抱书卷，木质发簪。",
|
||||
"灰白短发，凌厉眼神，左脸刺青，金属耳环。",
|
||||
"栗色长卷发，妩媚动人，红唇烈焰，玫瑰发饰。",
|
||||
"藏蓝束发，干练利落，目光如炬，黑色头巾。",
|
||||
"浅黄披发，病若西子，楚楚可怜，素色发带。",
|
||||
"七彩流光发，宛如神女，双眸含光，云雾缭绕。",
|
||||
"乌黑姬发式，乖巧文静，低眉顺眼，丝绸蝴蝶结。",
|
||||
"浅褐盘发，娇俏可人，插着糖葫芦，红绳发饰。",
|
||||
"银白长辫，圣洁高雅，闭目祈祷，柔和光环。",
|
||||
"深紫直发，冷艳高傲，下巴微扬，紫晶皇冠。",
|
||||
"翠绿双环髻，灵动活泼，手持折扇，翡翠流苏。",
|
||||
"铂金长发，冷漠疏离，如冰山雪莲，冰凌耳饰。",
|
||||
"蓬松红发，野性难驯，兽皮衣饰，骨牙项链。",
|
||||
"靛青垂鬟，知书达理，手持毛笔，书卷气息。",
|
||||
"绯红长发，傲娇神情，双手抱胸，猫耳发箍。",
|
||||
"墨蓝劲装，女扮男装，英姿飒爽，腰间佩剑。",
|
||||
"米色散发，睡眼惺忪，慵懒倚靠，云纹抱枕。",
|
||||
"黛青盘发，成熟稳重，慈眉善目，祥云发簪。",
|
||||
"碧绿长发，鬼气森森，面色苍白，幽冥鬼火。",
|
||||
"玫瑰金长发，公主气质，甜美微笑，蕾丝发带。",
|
||||
"漆黑长直，遮住单眼，阴郁神秘，骷髅发夹。",
|
||||
]
|
||||
male_prompt_base = "一个英俊的的仙侠男性头像。只有头部和面部。二次元风格的漫画图片，略微Q版，正面看镜头。纯白背景。像素风格，细节别太多。"
|
||||
male_prompt_base = "一个英俊的的仙侠男性头像。只有头部和面部且完整露出头部。二次元风格的漫画图片，略微Q版，正面看镜头。纯白背景。像素风格，细节别太多。"
|
||||
# male_affixes = [
|
||||
# "乌发高束，剑眉星目，气质冷峻，青玉发冠。",
|
||||
# "银白长发，淡笑从容，额间玄纹，流苏头箍。",
|
||||
# "墨发披肩，脸上一抹浅疤，坚毅沉稳，黑金发簪。",
|
||||
# "深棕短发，目光凌厉，薄唇紧抿，皮绳束发。",
|
||||
# "蓝黑长发，发尾微卷，温润如玉，白玉簪。",
|
||||
# "赤褐长发，桀骜挑眉，轻笑不羁，耳坠小铜铃。",
|
||||
# "玄青半束发，沉静内敛，额前碎发，银纹额饰。",
|
||||
# "白发如雪，清隽淡笑，眉心一点冰蓝印，细环头饰。",
|
||||
# "墨发高马尾，目如寒星，英气逼人，羽纹发冠。",
|
||||
# "亚麻色短发，随性浅笑，轻胡茬，细革头环。",
|
||||
# "乌青长发，神情冷淡，眼神专注，剑形耳坠。",
|
||||
# "银灰长直发，肃杀气质，额缠黑带，简洁利落。",
|
||||
# "深紫挑染长发，狡黠微笑，眸底流光，狐尾发饰。",
|
||||
# "墨发半披，眼神温和从容，玉串发夹。",
|
||||
# "金棕长发，爽朗大笑，额前碎发，兽牙发簪。",
|
||||
# "青黑短发，专注坚定，线条硬朗，细链发饰垂坠。",
|
||||
# ]
|
||||
male_affixes = [
|
||||
"乌发高束，剑眉星目，气质冷峻，青玉发冠。",
|
||||
"银白长发，淡笑从容，额间玄纹，流苏头箍。",
|
||||
"墨发披肩，脸上一抹浅疤，坚毅沉稳，黑金发簪。",
|
||||
"深棕短发，目光凌厉，薄唇紧抿，皮绳束发。",
|
||||
"蓝黑长发，发尾微卷，温润如玉，白玉簪。",
|
||||
"赤褐长发，桀骜挑眉，轻笑不羁，耳坠小铜铃。",
|
||||
"玄青半束发，沉静内敛，额前碎发，银纹额饰。",
|
||||
"白发如雪，清隽淡笑，眉心一点冰蓝印，细环头饰。",
|
||||
"墨发高马尾，目如寒星，英气逼人，羽纹发冠。",
|
||||
"亚麻色短发，随性浅笑，轻胡茬，细革头环。",
|
||||
"乌青长发，神情冷淡，眼神专注，剑形耳坠。",
|
||||
"银灰长直发，肃杀气质，额缠黑带，简洁利落。",
|
||||
"深紫挑染长发，狡黠微笑，眸底流光，狐尾发饰。",
|
||||
"墨发半披，眼神温和从容，玉串发夹。",
|
||||
"金棕长发，爽朗大笑，额前碎发，兽牙发簪。",
|
||||
"青黑短发，专注坚定，线条硬朗，细链发饰垂坠。",
|
||||
"墨黑长发，剑眉入鬓，眼神如电，束发金冠。",
|
||||
"银白散发，仙风道骨，捻须微笑，木质道簪。",
|
||||
"酒红短发，狂放不羁，嘴角轻挑，墨玉耳扣。",
|
||||
"深蓝马尾，冷若冰霜，目光锐利，银色护额。",
|
||||
"棕褐寸头，憨厚老实，笑容淳朴，粗布麻衣。",
|
||||
"金黄卷发，风流倜傥，桃花眼，折扇轻摇。",
|
||||
"灰白长发，面容枯槁，眼神阴鸷，骨质发饰。",
|
||||
"纯黑碎发，少年意气，眼神清澈，红色抹额。",
|
||||
"紫发披肩，妖异俊美，邪魅一笑，蛇形耳坠。",
|
||||
"青丝半束，书卷气息，温文尔雅，玉佩腰饰。",
|
||||
"赤发冲天，怒目圆睁，气势汹汹，火焰纹身。",
|
||||
"亚麻长发，忧郁深沉，低头沉思，素色发带。",
|
||||
"墨绿短发，干练果决，面无表情，单片琉璃镜。",
|
||||
"栗色微卷，温柔体贴，眼神宠溺，宽松道袍。",
|
||||
"藏蓝长辫，异域风情，肤色古铜，图腾面纹。",
|
||||
"浅灰中分，斯文败类，金边叆叇，嘴角冷笑。",
|
||||
"狂傲琴师，长发如瀑，抚琴长啸，音波缭绕。",
|
||||
"乌黑背头，一方霸主，目光审视，龙纹扳指。",
|
||||
"焦糖色蓬松，阳光开朗，露齿大笑，锦织抹额。",
|
||||
"银发遮眼，神秘莫测，嘴角微勾，面罩遮脸。",
|
||||
"深紫长直，高贵冷艳，不可一世，紫金皇冠。",
|
||||
"翠绿短发，灵动狡黠，吹着口哨，草叶衔嘴。",
|
||||
"铂金长发，正义凛然，手持长剑，玉石护符。",
|
||||
"棕红乱发，落魄浪人，胡渣唏嘘，酒葫芦。",
|
||||
"靛青束发，忠诚护卫，如影随形，蒙面黑巾。",
|
||||
"绯红短发，热血少年，握拳加油，脸颊伤痕。",
|
||||
"墨蓝狼尾，野性难驯，眼神凶狠，兽牙项链。",
|
||||
"米白长卷，慵懒贵族，品着灵茶，丝绸法袍。",
|
||||
"黛青道髻，严严肃穆，手持拂尘，八卦道袍。",
|
||||
"乱发蓬松，机关大师，佩戴透镜，摆弄零件。",
|
||||
"玫瑰金分头，花花公子，抛个媚眼，玫瑰花。",
|
||||
"漆黑长发，入魔之相，双目赤红，魔气缭绕。",
|
||||
]
|
||||
sect_prompt_base = "像素化的仙侠宗门场景图片，极度像素化，颗粒感强，线条轮廓粗，极简主义，二次元风格漫画图片。"
|
||||
sect_affixes = [
|
||||
|
||||
@@ -245,7 +245,7 @@ def process_all(
|
||||
if rename_by_index and saved_files:
|
||||
renamed_files: list[Path] = []
|
||||
for index, old_path in enumerate(saved_files, start=1):
|
||||
new_name = f"{index}.png"
|
||||
new_name = f"{index}_avatar.png"
|
||||
new_path = output_path / new_name
|
||||
old_path.rename(new_path)
|
||||
renamed_files.append(new_path)
|
||||
@@ -312,30 +312,30 @@ def process_all_sects(
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# process_all(
|
||||
# input_dir="tools/img_gen/tmp/males",
|
||||
# output_dir="tools/img_gen/tmp/processed_males",
|
||||
# crop_fraction=1 / 16,
|
||||
# )
|
||||
# process_all(
|
||||
# input_dir="tools/img_gen/tmp/females",
|
||||
# output_dir="tools/img_gen/tmp/processed_females",
|
||||
# crop_fraction=1 / 16,
|
||||
# )
|
||||
sect_names = [
|
||||
"明心剑宗",
|
||||
"百兽宗",
|
||||
"水镜宗",
|
||||
"冥王宗",
|
||||
"朱勾宗",
|
||||
"合欢宗",
|
||||
"镇魂宗",
|
||||
"幽魂噬影宗",
|
||||
"千帆城",
|
||||
]
|
||||
process_all_sects(
|
||||
input_dir="tools/img_gen/tmp/sects",
|
||||
output_dir="tools/img_gen/tmp/processed_sects",
|
||||
process_all(
|
||||
input_dir="tools/img_gen/tmp/males",
|
||||
output_dir="tools/img_gen/tmp/processed_males",
|
||||
crop_fraction=1 / 16,
|
||||
sect_names=sect_names,
|
||||
)
|
||||
)
|
||||
process_all(
|
||||
input_dir="tools/img_gen/tmp/females",
|
||||
output_dir="tools/img_gen/tmp/processed_females",
|
||||
crop_fraction=1 / 16,
|
||||
)
|
||||
# sect_names = [
|
||||
# "明心剑宗",
|
||||
# "百兽宗",
|
||||
# "水镜宗",
|
||||
# "冥王宗",
|
||||
# "朱勾宗",
|
||||
# "合欢宗",
|
||||
# "镇魂宗",
|
||||
# "幽魂噬影宗",
|
||||
# "千帆城",
|
||||
# ]
|
||||
# process_all_sects(
|
||||
# input_dir="tools/img_gen/tmp/sects",
|
||||
# output_dir="tools/img_gen/tmp/processed_sects",
|
||||
# crop_fraction=1 / 16,
|
||||
# sect_names=sect_names,
|
||||
# )
|
||||
Reference in New Issue
Block a user