0


人工智能AI系列 - java 版的stable diffusion 图像生成

图像生成

  • 文生图:输入提示词(仅支持英文),生成图片(仅支持英文)
  • GPU版本 StableDiffusionGPU.java
  • CPU版本 StableDiffusionCPU.java
  • 图生图:根据图片及提示词(仅支持英文)生成图片
  • GPU版本 Img2ImgStableDiffusionGPU.java
  • 显卡CUDA:11.7版本
  • 参考测试数据:分辨率 512*512 25步 CPU(i5处理器) 5分钟。 3060显卡20秒。

模型下载

  • cpu模型保存位置:stable_diffusion/pytorch_cpu
  • gpu模型保存位置:stable_diffusion/pytorch_gpu

文生图测试

  • 提示词 prompt: a photo of an astronaut riding a horse on mars
  • 反向提示词 negative_prompt:
  • 生成图片效果:

图像生成提示词参考

作品欣赏

SDK代码下载地址:

Github链接
Gitee链接


本文转载自: https://blog.csdn.net/weixin_39355136/article/details/129459687
版权归原作者 Calvin.AIAS 所有, 如有侵权,请联系我们删除。

“人工智能AI系列 - java 版的stable diffusion 图像生成”的评论:

还没有评论