簡介
waifu-diffusion的模型可用于生成二次元的卡通形象,趕快來生成你的二次元動漫小姐姐和主人公,
- waifu-diffusion教程文檔:https://github.com/PaddlePaddle/PaddleHub/tree/develop/modules/image/text_to_image
安裝并更新PaddleHub
!pip install --upgrade paddlehub
複制代碼
加載模型
import paddlehub as hub
複制代碼
模型推理
text_prompts = ["Ayusawa Misaki"]
複制代碼
da = module.generate_image(text_prompts=text_prompts, batch_size=3, output_dir='./stable_diffusion_out/')
# 展示所有的中間結果
da[0].chunks[-1].chunks.plot_image_sprites(skip_empty=True, show_index=True, keep_aspect_ratio=True)
# 将整個生成過程儲存為一個動态圖gif
da[0].chunks[-1].chunks.save_gif('stable_diffusion_waifu_out-merged-result.gif')
# da索引的是prompt, da[0].chunks索引的是該prompt下生成的第一張圖,在batch_size不為1時能同時生成多張圖
# 您也可以按照上述操作顯示單張圖,如第0張的生成過程
da[0].chunks[0].chunks.plot_image_sprites(skip_empty=True, show_index=True, keep_aspect_ratio=True)
da[0].chunks[0].chunks.save_gif('stable_diffusion_waifu_out-image-0-result.gif')
複制代碼
模型推理
text_prompts = ["Ayusawa Misaki"]
複制代碼
da = module.generate_image(text_prompts=text_prompts, batch_size=3, output_dir='./stable_diffusion_out/')
# 展示所有的中間結果
da[0].chunks[-1].chunks.plot_image_sprites(skip_empty=True, show_index=True, keep_aspect_ratio=True)
# 将整個生成過程儲存為一個動态圖gif
da[0].chunks[-1].chunks.save_gif('stable_diffusion_waifu_out-merged-result.gif')
# da索引的是prompt, da[0].chunks索引的是該prompt下生成的第一張圖,在batch_size不為1時能同時生成多張圖
# 您也可以按照上述操作顯示單張圖,如第0張的生成過程
da[0].chunks[0].chunks.plot_image_sprites(skip_empty=True, show_index=True, keep_aspect_ratio=True)
da[0].chunks[0].chunks.save_gif('stable_diffusion_waifu_out-image-0-result.gif')
複制代碼