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]