patrickvonplaten commited on
Commit
ea474bd
·
1 Parent(s): 8335892

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -1
README.md CHANGED
@@ -1,4 +1,40 @@
1
  ---
2
  tags:
3
  - ddpm_diffusion
4
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  tags:
3
  - ddpm_diffusion
4
+ ---
5
+
6
+ # Denoising Diffusion Probabilistic Models (DDPM)
7
+
8
+ **Paper**: [Denoising Diffusion Probabilistic Models](https://arxiv.org/abs/2006.11239)
9
+
10
+ **Abstract**:
11
+
12
+ *We present high quality image synthesis results using diffusion probabilistic models, a class of latent variable models inspired by considerations from nonequilibrium thermodynamics. Our best results are obtained by training on a weighted variational bound designed according to a novel connection between diffusion probabilistic models and denoising score matching with Langevin dynamics, and our models naturally admit a progressive lossy decompression scheme that can be interpreted as a generalization of autoregressive decoding. On the unconditional CIFAR10 dataset, we obtain an Inception score of 9.46 and a state-of-the-art FID score of 3.17. On 256x256 LSUN, we obtain sample quality similar to ProgressiveGAN.*
13
+
14
+ ## Usage
15
+
16
+ ```python
17
+ from diffusers import DiffusionPipeline
18
+ import PIL.Image
19
+ import numpy as np
20
+
21
+ model_id = "fusing/ddpm-lsun-bedroom"
22
+
23
+ # load model and scheduler
24
+ ddpm = DiffusionPipeline.from_pretrained(model_id)
25
+
26
+ # run pipeline in inference (sample random noise and denoise)
27
+ image = ddpm()
28
+
29
+ # process image to PIL
30
+ image_processed = image.cpu().permute(0, 2, 3, 1)
31
+ image_processed = (image_processed + 1.0) * 127.5
32
+ image_processed = image_processed.numpy().astype(np.uint8)
33
+ image_pil = PIL.Image.fromarray(image_processed[0])
34
+
35
+ # save image
36
+ image_pil.save("test.png")
37
+ ```
38
+
39
+ ## Samples
40
+