piggyback-color

用文本提示给黑白照片智能上色，效果自然可控

piggyback-color 是一个基于扩散模型的图像着色项目，通过“piggybacked models”（附加模型）的方式，在预训练的 Stable Diffusion 模型上叠加轻量级分支，实现文本引导的交互式图像着色。它解决了传统着色方法需要大量人工标注或结果不自然的问题，利用扩散模型的强大生成能力，将灰度图像自动着色，并支持用户通过文本描述或颜色提示来控制着色结果。核心能力包括：利用 Stable Diffusion 的语义理解生成符合内容逻辑的颜色；支持局部交互式着色，用户可指定区域和颜色；提供训练和推理代码，便于研究者复现和改进。项目处于早期阶段，代码和文档尚在完善中。