add dps config

diffusion-posterior-sampling/configs/box_inpainting_config.yaml

@@ -0,0 +1,22 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.5
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: inpainting
+  mask_opt:
+    mask_type: box
+    mask_len_range: !!python/tuple [127, 128]  # for box
+    # mask_prob_range: !!python/tuple [0.3, 0.7]  # for random
+    image_size: 256
+
+
+  noise:
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/box_inpainting_config_psld.yaml

@@ -0,0 +1,22 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.5
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: inpainting
+  mask_opt:
+    mask_type: box
+    mask_len_range: !!python/tuple [256, 257]  # for box
+    # mask_prob_range: !!python/tuple [0.3, 0.7]  # for random
+    image_size: 512
+
+
+  noise:
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/diffusion_config.yaml

@@ -0,0 +1,9 @@
+sampler: ddpm
+steps: 1000
+noise_schedule: linear
+model_mean_type: epsilon 
+model_var_type: learned_range
+dynamic_threshold: False
+clip_denoised: True
+rescale_timesteps: False
+timestep_respacing: 1000

diffusion-posterior-sampling/configs/gaussian_deblur_config.yaml

@@ -0,0 +1,18 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: gaussian_blur
+    kernel_size: 61
+    intensity: 3.0
+  
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/gaussian_deblur_config_psld.yaml

@@ -0,0 +1,18 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: gaussian_blur
+    kernel_size: 121
+    intensity: 3.0
+  
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/imagenet_model_config.yaml

@@ -0,0 +1,20 @@
+# Defaults for image training.
+
+image_size: 256
+num_channels: 256
+num_res_blocks: 2
+channel_mult: ""
+learn_sigma: True
+class_cond: False
+use_checkpoint: False
+attention_resolutions: "32,16,8"
+num_heads: 4
+num_head_channels: 64
+num_heads_upsample: -1
+use_scale_shift_norm: True
+dropout: 0.0
+resblock_updown: True
+use_fp16: False
+use_new_attention_order: False
+
+model_path: models/imagenet256.pt

diffusion-posterior-sampling/configs/inpainting_config.yaml

@@ -0,0 +1,22 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.5
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: inpainting
+  mask_opt:
+    mask_type: random
+    # mask_len_range: !!python/tuple [128, 129]  # for box
+    mask_prob_range: !!python/tuple [0.3, 0.7]  # for random
+    image_size: 256
+
+
+  noise:
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/inpainting_config_psld.yaml

@@ -0,0 +1,22 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.5
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: inpainting
+  mask_opt:
+    mask_type: random
+    # mask_len_range: !!python/tuple [128, 129]  # for box
+    mask_prob_range: !!python/tuple [0.3, 0.7]  # for random
+    image_size: 512
+
+
+  noise:
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/model_config.yaml

@@ -0,0 +1,20 @@
+# Defaults for image training.
+
+image_size: 256
+num_channels: 128
+num_res_blocks: 1
+channel_mult: ""
+learn_sigma: True
+class_cond: False
+use_checkpoint: False
+attention_resolutions: 16
+num_heads: 4
+num_head_channels: 64
+num_heads_upsample: -1
+use_scale_shift_norm: True
+dropout: 0.0
+resblock_updown: True
+use_fp16: False
+use_new_attention_order: False
+
+model_path: models/ffhq_10m.pt

diffusion-posterior-sampling/configs/motion_deblur_config.yaml

@@ -0,0 +1,18 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: motion_blur
+    kernel_size: 61
+    intensity: 0.5
+
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/motion_deblur_config_psld.yaml

@@ -0,0 +1,18 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: motion_blur
+    kernel_size: 121
+    intensity: 0.5
+
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/nonlinear_deblur_config.yaml

@@ -0,0 +1,17 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: nonlinear_blur
+    opt_yml_path: ./bkse/options/generate_blur/default.yml
+
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/phase_retrieval_config.yaml

@@ -0,0 +1,17 @@
+conditioning:
+  method: ps
+  params:
+    scale: 1.0
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: phase_retrieval
+    oversample: 2.0
+
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/super_resolution_config.yaml

@@ -0,0 +1,18 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: super_resolution
+    in_shape: !!python/tuple [1, 3, 256, 256]
+    scale_factor: 4
+  
+  noise: 
+    name: gaussian
+    sigma: 0.05

diffusion-posterior-sampling/configs/super_resolution_config_psld.yaml

@@ -0,0 +1,18 @@
+conditioning:
+  method: ps
+  params:
+    scale: 0.3
+
+data:
+  name: ffhq
+  root: ./data/samples/
+
+measurement:
+  operator:
+    name: super_resolution
+    in_shape: !!python/tuple [1, 3, 512, 512]
+    scale_factor: 4
+  
+  noise: 
+    name: gaussian
+    sigma: 0.05