tiny-dream
轻量级 Stable Diffusion C++ 推理库
Tiny Dream 是一个轻量级的 Stable Diffusion C++ 实现，仅包含头文件且无外部依赖。该库专注于提高 CPU 效率和减少内存占用，在普通硬件上即可高效运行。Tiny Dream 内存需求低，支持多种高级功能，并提供简洁的 C++ API，便于集成到现有项目中。
Tiny DreamStable DiffusionC++嵌入式图像生成Github开源项目
<h1 align="center">微梦境<br><br>一个嵌入式、仅头文件的稳定扩散推理C++库<br><a href="https://pixlab.io/tiny-dream">pixlab.io/tiny-dream</a></h1>
#include "tinydream.hpp"
/*
* 主入口点。唯一必需的参数是正向提示词。
* 不过，强烈建议也传入负向提示词（用逗号分隔的词语）。
*
* 我们建议您尝试不同的种子值和步数，
* 以达到理想的结果。
*
* ./tinydream "正向提示词" ["负向提示词"] [种子值] [步数]
*/
int main(int argc, char *argv[])
{
tinyDream td; // 栈上分配的tinyDream对象
// 显示库当前的推理引擎、版本号和版权声明
std::cout << tinyDream::about() << std::endl;
// 至少需要通过命令行提供一个正面提示
if (argc < 2) {
std::cout << "缺少正面（可能还有负面）提示：请描述你想要生成的内容..." << std::endl;
std::cout << "提示示例：" << std::endl;
// 内置正面/负面提示的示例
auto prompts = tinyDream::promptExample();
std::cout << "\t正面提示：" << prompts.first << std::endl;
std::cout << "\t负面提示：" << prompts.second << std::endl;
return -1;
}
// 注册负责处理推理过程中生成的日志消息的回调函数
td.setLogCallback(logCallback, nullptr);
// 可选：如果预训练模型不在可执行文件所在目录，请设置资源路径
// Tiny-Dream资源可以从以下地址下载：https://pixlab.io/tiny-dream#downloads
td.setAssetsPath("/path/to/tinydream/assets"); // 如果资源位于可执行文件所在目录，请删除或注释此行
// 可选：为每个新生成的图像名称设置前缀
td.setImageOutputPrefix("tinydream-");
// 可选：设置生成图像的存储目录
td.setImageOutputPath("/home/photos/");
int seedMax = 90;
if (argc > 3) {
/*
* Stable Diffusion中的种子是用于初始化生成过程的数字。
* 控制种子可以帮助你生成可重复的图像，实验其他参数，或提示变化。
*/
seedMax = std::atoi(argv[3]);
}
int step = 30;
if (argc > 4) {
/*
* 调整Stable Diffusion的推理步骤：使用的步骤越多，
* 质量越好，但不应将步骤设置得尽可能高。
* 大约30个采样步骤（默认值）通常足以生成高质量图像。
*/
step = std::atoi(argv[4]);
}
/*
* 用户提供的提示 - 生成符合输入条件的图像。
*
* 正面提示（必需）：描述你想要生成的内容（逗号分隔的词语）。
* 负面提示（可选）：描述你不想看到的内容（逗号分隔的词语）。
*/
std::string positivePrompt{ argv[1] };
std::string negativePrompt{ "" };
if (argc > 2) {
negativePrompt = std::string{ argv[2] };
}
/*
* 最后，运行Stable Diffusion进行推理
*
* 之前注册的日志消费者回调函数应该很快就会收到
* 推理过程中生成的所有日志消息（包括错误，如果有的话）。
*
* 关于tinyDream::dream()方法所需参数的说明，
* 请参考官方文档：https://pixlab.io/tiny-dream#tiny-dream-method
*/
for (int seed = 1; seed < seedMax; seed++) {
std::string outputImagePath;
td.dream(
positivePrompt,
negativePrompt,
outputImagePath,
true, /* 如果你想要512x512像素的输出而不是2048x2048的输出，请设置为false */
seed,
step
);
}
// 你不需要每次都通过std::cout手动显示生成的图像路径
// 因为提供的日志回调应该已经完成了这项工作。
std::cout << "输出图像位置: " << outputImagePath << std::endl; // 如果太繁琐可以取消注释
}
return 0;
}
编辑推荐精选
讯飞智文
一键生成PPT和Word，让学习生活更轻松
讯飞智文是一个利用 AI 技术的项目，能够帮助用户生成 PPT 以及各类文档。无论是商业领域的市场分析报告、年度目标制定，还是学生群体的职业生涯规划、实习避坑指南，亦或是活动策划、旅游攻略等内容，它都能提供支持，帮助用户精准表达，轻松呈现各种信息。
AI办公办公工具AI工具讯飞智文AI在线生成PPTAI撰写助手多语种文档生成AI自动配图热门
讯飞星火
深度推理能力全新升级，全面对标OpenAI o1
科大讯飞的星火大模型，支持语言理解、知识问答和文本创作等多功能，适用于多种文件和业务场景，提升办公和日常生活的效率。讯飞星火是一个提供丰富智能服务的平台，涵盖科技资讯、图像创作、写作辅助、编程解答、科研文献解读等功能，能为不同需求的用户提供便捷高效的帮助，助力用户轻松获取信息、解决问题，满足多样化使用场景。
热门AI开发模型训练AI工具讯飞星火大模型智能问答内容创作多语种支持智慧生活
Spark-TTS
一种基于大语言模型的高效单流解耦语音令牌文本到语音合成模型
Spark-TTS 是一个基于 PyTorch 的开源文本到语音合成项目，由多个知名机构联合参与。该项目提供了高效的 LLM（大语言模型）驱动的语音合成方案，支持语音克隆和语音创建功能，可通过命令行界面（CLI）和 Web UI 两种方式使用。用户可以根据需求调整语音的性别、音高、速度等参数，生成高质量的语音。该项目适用于多种场景，如有声读物制作、智能语音助手开发等。
Trae
字节跳动发布的AI编程神器IDE
Trae是一种自适应的集成开发环境（IDE），通过自动化和多元协作改变开发流程。利用Trae，团队能够更快速、精确地编写和部署代码，从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能，是提升开发效率的理想工具。
AI工具TraeAI IDE协作生产力转型热门
咔片PPT
AI助力，做PPT更简单！
咔片是一款轻量化在线演示设计工具，借助 AI 技术，实现从内容生成到智能设计的一站式 PPT 制作服务。支持多种文档格式导入生成 PPT，提供海量模板、智能美化、素材替换等功能，适用于销售、教师、学生等各类人群，能高效制作出高品质 PPT，满足不同场景演示需求。
讯飞绘文
选题、配图、成文，一站式创作，让内容运营更高效
讯飞绘文，一个AI集成平台，支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容，加速品牌传播 ，提升内容营销效果。
热门AI辅助写作AI工具讯飞绘文内容运营AI创作个性化文章多平台分发AI助手
材料星
专业的AI公文写作平台，公文写作神器
AI 材料星，专业的 AI 公文写作辅助平台，为体制内工作人员提供高效的公文写作解决方案。拥有海量公文文库、9 大核心 AI 功能，支持 30 + 文稿类型生成，助力快速完成领导讲话、工作总结、述职报告等材料，提升办公效率，是体制打工人的得力写作神器。
openai-agents-python
OpenAI Agents SDK，助力开发者便捷使用 OpenAI 相关功能。
openai-agents-python 是 OpenAI 推出的一款强大 Python SDK，它为开发者提供了与 OpenAI 模型交互的高效工具，支持工具调用、结果处理、追踪等功能，涵盖多种应用场景，如研究助手、财务研究等，能显著提升开发效率，让开发者更轻松地利用 OpenAI 的技术优势。
Hunyuan3D-2
高分辨率纹理 3D 资产生成
Hunyuan3D-2 是腾讯开发的用于 3D 资产生成的强大工具，支持从文本描述、单张图片或多视角图片生成 3D 模型，具备快速形状生成能力，可生成带纹理的高质量 3D 模型，适用于多个领域，为 3D 创作提供了高效解决方案。
3FS
一个具备存储、管理和客户端操作等多种功能的分布式文件系统相关项目。
3FS 是一个功能强大的分布式文件系统项目，涵盖了存储引擎、元数据管理、客户端工具等多个模块。它支持多种文件操作，如创建文件和目录、设置布局等，同时具备高效的事件循环、节点选择和协程池管理等特性。适用于需要大规模数据存储和管理的场景，能够提高系统的性能和可靠性，是分布式存储领域的优质解决方案。
下拉加载更多
推荐工具精选
AI云服务特惠
懂AI专属折扣关注微信公众号
最新AI工具、AI资讯
独家AI资源、AI项目落地
微信扫一扫关注公众号