Skip to content

Rectified Flow with Conditional Input for MNIST Image Generation

Notifications You must be signed in to change notification settings

whizhou/Rectified-Flow-MNIST

Repository files navigation

Flow Matching (Rectified Flow) 处理 Mnist 数据集

使用 Unet 作为核心实现 Mnist 数据集的随机或条件生成。

结果说明

  • v1.1版本同时支持无条件生成和条件生成,共训练50个epoch。

  • 模型收敛较好

loss_curve

epoch 50, step 10, conditional

result_img

epoch 50, step 10, unconditional

result_img

About

Rectified Flow with Conditional Input for MNIST Image Generation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages