加法扩散模型全部过程推导和实现代码

*★,°*:.☆欢迎您/$:*.°★*

---

---

add_diff 使用之前的扩散方法(get_image_by_t_cv)总结出来的
get_noise  和 get_x 通过 add_diff 可以得到
通过get_x  带入 两组参数  可以推导出 get_xt_1
使用add_diff(这里要反着用 参考show_add_diff_r) 输入 x noise t t max 可以得到xt
 xt 和 t 输入网络中得到 pre_noise 和noise 做 loss
 推理过程  随机使用一个noise=Xt  和 t
 输入到网络中 得到 pre_noise  随机一个new_noise
 将xt pre_noise new_noise t 输入到 get_xt_1
 得到 xt_1  xt=xt_1 t=t+1
 重复

import numpy as np
import cv2
import matplotlib.pyplot as plt


def get