Tiny Dream: 一个嵌入式、仅头文件的稳定扩散C++实现

tiny-dream

Tiny Dream: 小巧精致的稳定扩散实现

在人工智能和机器学习领域,稳定扩散(Stable Diffusion)技术的出现无疑是一个重大突破。它能够根据文本描述生成高质量的图像,为创意工作者提供了强大的工具。然而,传统的稳定扩散模型通常需要庞大的计算资源和复杂的环境配置,这在某些场景下可能会成为应用的障碍。为了解决这一问题,PixLab公司开发了Tiny Dream - 一个小巧精致、高效实用的稳定扩散C++实现。

Tiny Dream的核心特性

Tiny Dream作为一个轻量级的稳定扩散实现,具有以下突出特点:

仅头文件实现: Tiny Dream的整个代码库仅由两个头文件组成 - tinydream.hpp和stb_image_write.h。这种设计使得它极易集成到现有的C++项目中。
无外部依赖: 除了用于保存图像的stb_image_write.h外,Tiny Dream不需要其他任何外部库。这大大简化了部署和使用过程。
低内存占用: Tiny Dream在运行时仅需1.7~5.5GB的内存,这在同类实现中是相当优秀的表现。
CPU友好: 虽然不强制要求NVIDIA GPU,但Tiny Dream在普通的CPU上也能以合理的速度运行。它利用了TBB线程和SSE/AVX矢量化技术来提升性能。
易于嵌入: 简洁的C++ API使得Tiny Dream可以轻松地嵌入到更大的代码库中。

Tiny Dream示例图

使用Tiny Dream

将Tiny Dream集成到现有项目中非常简单。以下是基本步骤:

下载Tiny Dream的源代码和预训练模型。
将tinydream.hpp和stb_image_write.h添加到您的项目中。
实例化一个tinyDream对象。
使用dream()方法生成图像。

以下是一个简单的示例代码:

#include "tinydream.hpp"

int main() {
    tinyDream td;
    td.setAssetsPath("/path/to/tinydream/assets");
    
    std::string outputImagePath;
    td.dream(
        "一座金字塔,沙漠,棕榈树,河流,(景观),(高质量)",
        "模糊,低质量",
        outputImagePath,
        true,
        42,
        30
    );
    
    std::cout << "生成的图像路径: " << outputImagePath << std::endl;
    return 0;
}

这个简单的例子展示了如何使用Tiny Dream生成一张描述沙漠景观的图像。