Diffusion Aurora 256

🚀 极光扩散模型

本项目利用生成模型生成极光图片，解决了因极光罕见难以亲眼目睹的问题，为用户提供了一种欣赏极光之美的新途径。

数据集与标签

项目展示

🚀 快速开始

模型描述

你是否曾亲眼见过极光？看看上面这张我冬天在阿拉斯加拍摄的照片。很美，对吧？然而，极光十分罕见，即使在阿拉斯加这样极北的地方，我们也很难见到。别担心，现在我们有生成模型啦！以下是该模型生成的图片：

模型细节

此模型可生成 256 * 256 像素的极光图片。它基于数据集 huggan/few-shot-aurora 进行训练。训练方法参考了这个 示例。你可以在这里查看我的训练源代码：

使用示例

基础用法

from diffusers import DDPMPipeline

pipeline = DDPMPipeline.from_pretrained('li-yan/diffusion-aurora-256')
image = pipeline().images[0]
image

高级用法

from diffusers import DiffusionPipeline, DDIMScheduler

scheduler = DDIMScheduler.from_pretrained('li-yan/diffusion-aurora-256')
scheduler.set_timesteps(num_inference_steps=40)

pipeline = DiffusionPipeline.from_pretrained(
    'li-yan/diffusion-aurora-256', scheduler=scheduler)

images = pipeline(num_inference_steps=40).images
images[0]