扩散模型中的prediction_type详解

最近在学习Diffusion,今天在coding的过程中,看到了一个参数prediction_type,目前代码里面是有三种类型,分别是sampleepsolonv_prediction,在好奇心的驱使下,我想要对prediction_type一探究竟,在这里也顺便记录一下。

什么是prediction_type

prediction_type是扩散模型中调度器(scheduler)的一个重要参数。它决定了模型在训练和推理过程中如何预测噪声或潜在变量,不同的预测类型会影响模型的性能和生成效果。

prediction_type的类型以及作用

epsilon

作用:epsilon类型是直接预测添加到原始图像中的噪声。模型学习预测在扩散过程中每个时间步添加到图像中的噪声值。在去噪过程中,根据模型预测的噪声来逐步去除噪声,恢复原始图像。
通常应用于传统的扩散模型,如原始的DDPM(Denoising Diffusion Probabilistic Models)模型。它是一种较为直观的预测方式,通过预测噪声的方式来逼近真实的无噪声图像。

举例:在Stable Diffusion v1.x模型中,通常使用epsilon作为预测类型。例如,当使用预训练的Stable Diffusion v1.x模型进行图像生成时,模型的调度器会根据epsilon

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

程序员非鱼

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值