├── LICENSE ├── README.md ├── ScoreFPE_3Doutline_single.gif ├── Score_SDE_demo_PyTorch.ipynb ├── configs ├── default_celeba_configs.py ├── default_cifar10_configs.py ├── default_lsun_configs.py ├── subvp │ ├── cifar10_ddpm_continuous.py │ ├── cifar10_ddpmpp_continuous.py │ ├── cifar10_ddpmpp_deep_continuous.py │ ├── cifar10_ncsnpp_continuous.py │ └── cifar10_ncsnpp_deep_continuous.py ├── ve │ ├── bedroom_ncsnpp_continuous.py │ ├── celeba_ncsnpp.py │ ├── celebahq_256_ncsnpp_continuous.py │ ├── celebahq_ncsnpp_continuous.py │ ├── church_ncsnpp_continuous.py │ ├── cifar10_ddpm.py │ ├── cifar10_ncsnpp.py │ ├── cifar10_ncsnpp_continuous.py │ ├── cifar10_ncsnpp_deep_continuous.py │ ├── ffhq_256_ncsnpp_continuous.py │ ├── ffhq_ncsnpp_continuous.py │ ├── ncsn │ │ ├── celeba.py │ │ ├── celeba_124.py │ │ ├── celeba_1245.py │ │ ├── celeba_5.py │ │ ├── cifar10.py │ │ ├── cifar10_124.py │ │ ├── cifar10_1245.py │ │ └── cifar10_5.py │ └── ncsnv2 │ │ ├── bedroom.py │ │ ├── celeba.py │ │ └── cifar10.py └── vp │ ├── cifar10_ddpmpp.py │ ├── cifar10_ddpmpp_continuous.py │ ├── cifar10_ddpmpp_deep_continuous.py │ ├── cifar10_ncsnpp.py │ ├── cifar10_ncsnpp_continuous.py │ ├── cifar10_ncsnpp_deep_continuous.py │ └── ddpm │ ├── bedroom.py │ ├── celebahq.py │ ├── church.py │ ├── cifar10.py │ ├── cifar10_continuous.py │ └── cifar10_unconditional.py ├── controllable_generation.py ├── datasets.py ├── debug.py ├── diff.py ├── evaluation.py ├── likelihood.py ├── losses.py ├── main.py ├── models ├── __init__.py ├── ddpm.py ├── ema.py ├── layers.py ├── layerspp.py ├── ncsnpp.py ├── ncsnv2.py ├── normalization.py ├── up_or_down_sampling.py └── utils.py ├── op ├── __init__.py ├── fused_act.py ├── fused_bias_act.cpp ├── fused_bias_act_kernel.cu ├── upfirdn2d.cpp ├── upfirdn2d.py └── upfirdn2d_kernel.cu ├── requirements.txt ├── run_lib.py ├── sampling.py ├── sde_lib.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /ScoreFPE_3Doutline_single.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/ScoreFPE_3Doutline_single.gif -------------------------------------------------------------------------------- /Score_SDE_demo_PyTorch.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/Score_SDE_demo_PyTorch.ipynb -------------------------------------------------------------------------------- /configs/default_celeba_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/default_celeba_configs.py -------------------------------------------------------------------------------- /configs/default_cifar10_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/default_cifar10_configs.py -------------------------------------------------------------------------------- /configs/default_lsun_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/default_lsun_configs.py -------------------------------------------------------------------------------- /configs/subvp/cifar10_ddpm_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/subvp/cifar10_ddpm_continuous.py -------------------------------------------------------------------------------- /configs/subvp/cifar10_ddpmpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/subvp/cifar10_ddpmpp_continuous.py -------------------------------------------------------------------------------- /configs/subvp/cifar10_ddpmpp_deep_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/subvp/cifar10_ddpmpp_deep_continuous.py -------------------------------------------------------------------------------- /configs/subvp/cifar10_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/subvp/cifar10_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/subvp/cifar10_ncsnpp_deep_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/subvp/cifar10_ncsnpp_deep_continuous.py -------------------------------------------------------------------------------- /configs/ve/bedroom_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/bedroom_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/celeba_ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/celeba_ncsnpp.py -------------------------------------------------------------------------------- /configs/ve/celebahq_256_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/celebahq_256_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/celebahq_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/celebahq_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/church_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/church_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/cifar10_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/cifar10_ddpm.py -------------------------------------------------------------------------------- /configs/ve/cifar10_ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/cifar10_ncsnpp.py -------------------------------------------------------------------------------- /configs/ve/cifar10_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/cifar10_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/cifar10_ncsnpp_deep_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/cifar10_ncsnpp_deep_continuous.py -------------------------------------------------------------------------------- /configs/ve/ffhq_256_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ffhq_256_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/ffhq_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ffhq_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/ve/ncsn/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/celeba.py -------------------------------------------------------------------------------- /configs/ve/ncsn/celeba_124.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/celeba_124.py -------------------------------------------------------------------------------- /configs/ve/ncsn/celeba_1245.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/celeba_1245.py -------------------------------------------------------------------------------- /configs/ve/ncsn/celeba_5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/celeba_5.py -------------------------------------------------------------------------------- /configs/ve/ncsn/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/cifar10.py -------------------------------------------------------------------------------- /configs/ve/ncsn/cifar10_124.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/cifar10_124.py -------------------------------------------------------------------------------- /configs/ve/ncsn/cifar10_1245.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/cifar10_1245.py -------------------------------------------------------------------------------- /configs/ve/ncsn/cifar10_5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsn/cifar10_5.py -------------------------------------------------------------------------------- /configs/ve/ncsnv2/bedroom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsnv2/bedroom.py -------------------------------------------------------------------------------- /configs/ve/ncsnv2/celeba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsnv2/celeba.py -------------------------------------------------------------------------------- /configs/ve/ncsnv2/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/ve/ncsnv2/cifar10.py -------------------------------------------------------------------------------- /configs/vp/cifar10_ddpmpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/cifar10_ddpmpp.py -------------------------------------------------------------------------------- /configs/vp/cifar10_ddpmpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/cifar10_ddpmpp_continuous.py -------------------------------------------------------------------------------- /configs/vp/cifar10_ddpmpp_deep_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/cifar10_ddpmpp_deep_continuous.py -------------------------------------------------------------------------------- /configs/vp/cifar10_ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/cifar10_ncsnpp.py -------------------------------------------------------------------------------- /configs/vp/cifar10_ncsnpp_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/cifar10_ncsnpp_continuous.py -------------------------------------------------------------------------------- /configs/vp/cifar10_ncsnpp_deep_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/cifar10_ncsnpp_deep_continuous.py -------------------------------------------------------------------------------- /configs/vp/ddpm/bedroom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/ddpm/bedroom.py -------------------------------------------------------------------------------- /configs/vp/ddpm/celebahq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/ddpm/celebahq.py -------------------------------------------------------------------------------- /configs/vp/ddpm/church.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/ddpm/church.py -------------------------------------------------------------------------------- /configs/vp/ddpm/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/ddpm/cifar10.py -------------------------------------------------------------------------------- /configs/vp/ddpm/cifar10_continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/ddpm/cifar10_continuous.py -------------------------------------------------------------------------------- /configs/vp/ddpm/cifar10_unconditional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/configs/vp/ddpm/cifar10_unconditional.py -------------------------------------------------------------------------------- /controllable_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/controllable_generation.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/datasets.py -------------------------------------------------------------------------------- /debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/debug.py -------------------------------------------------------------------------------- /diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/diff.py -------------------------------------------------------------------------------- /evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/evaluation.py -------------------------------------------------------------------------------- /likelihood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/likelihood.py -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/losses.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/ddpm.py -------------------------------------------------------------------------------- /models/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/ema.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/layers.py -------------------------------------------------------------------------------- /models/layerspp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/layerspp.py -------------------------------------------------------------------------------- /models/ncsnpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/ncsnpp.py -------------------------------------------------------------------------------- /models/ncsnv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/ncsnv2.py -------------------------------------------------------------------------------- /models/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/normalization.py -------------------------------------------------------------------------------- /models/up_or_down_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/up_or_down_sampling.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/models/utils.py -------------------------------------------------------------------------------- /op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/__init__.py -------------------------------------------------------------------------------- /op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/fused_act.py -------------------------------------------------------------------------------- /op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/upfirdn2d.py -------------------------------------------------------------------------------- /op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/run_lib.py -------------------------------------------------------------------------------- /sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/sampling.py -------------------------------------------------------------------------------- /sde_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/sde_lib.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/fp-diffusion/HEAD/utils.py --------------------------------------------------------------------------------