当前位置: 代码网 > it编程>编程语言>Java > 人工智能AI系列 - java 版的stable diffusion 图像生成

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

2024年07月28日 Java 我要评论
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链接

(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com