Improved Diffusion

IDDPM

diffusion 相比gan優(yōu)勢(shì)

attention unet

字典擴(kuò)充

add_dict_to_argparse

自動(dòng)化定義

從字典中自動(dòng)生成argument parser 很方便，不用一個(gè)個(gè)手打

Load data

list_image_files_recursively(data_dir)循環(huán)遞歸找圖片

ImageDataset

trainloop

create_model_and_diffusion

高斯diffusion加噪法

space diffusion

高斯diffusion