home diffusion ddpm
View code on Github
これは、論文「ノイズ除去拡散確率モデル」のPyTorch実装/チュートリアルです。
簡単に言うと、データから画像を取得し、段階的にノイズを追加します。次に、モデルをトレーニングして各ステップでそのノイズを予測し、そのモデルを使用して画像を生成します。
ノイズとトレーニングコードを予測する uNet モデルを次に示します。このファイルでは、トレーニング済みのモデルからサンプルと補間を生成できます