add gaussianblur

evals.py

@@ -36,7 +36,8 @@ DEFAULT_MODEL_PARAMS = {
 
 class PhysicsWithGenerator(torch.nn.Module):
     """Interface between Physics, Generator and Gradio."""
-    all_physics = ["MotionBlur_easy", "MotionBlur_medium", "MotionBlur_hard", "GaussianBlur",
+    all_physics = ["MotionBlur_easy", "MotionBlur_medium", "MotionBlur_hard",
+                   "GaussianBlur_easy", "GaussianBlur_medium", "GaussianBlur_hard",
                    "MRI", "CT"]
 
     def __init__(self, physics_name: str, device_str: str = "cpu") -> None:
@@ -77,16 +78,44 @@ class PhysicsWithGenerator(torch.nn.Module):
                                  "updatable_params_converter": {"sigma": float},
                                  "fixed_params": {"noise_sigma_min": 0.1, "noise_sigma_max": 0.1,
                                                   "psf_size": 31, "motion_gen_l": 1.2, "motion_gen_s": 1.0}}
-        elif self.name == "GaussianBlur":
+        elif self.name == "GaussianBlur_easy":
+            psf_size = 31
+            self.physics = dinv.physics.Blur(noise_model=dinv.physics.GaussianNoise(sigma=0.01), padding="valid",
+                                             device=device_str)
+            self.physics_generator = GaussianBlurGenerator(psf_size=(psf_size, psf_size),
+                                                           sigma_min=1.0, sigma_max=1.0,
+                                                           num_channels=1,
+                                                           device=device_str) + SigmaGenerator(sigma_min=0.01, sigma_max=0.01, device=device_str)
+            self.generator = self.physics_generator
+            self.saved_params = {"updatable_params": {},
+                                 "updatable_params_converter": {},
+                                 "fixed_params": {"noise_sigma": 0.01, "blur_sigma": 1.0,
+                                                  "psf_size": 31, "num_channels": 1}}
+        elif self.name == "GaussianBlur_medium":
             psf_size = 31
             self.physics = dinv.physics.Blur(noise_model=dinv.physics.GaussianNoise(sigma=0.05), padding="valid",
                                              device=device_str)
-            self.physics_generator = GaussianBlurGenerator(psf_size=(psf_size, psf_size), num_channels=1,
-                                                           device=device_str)
-            self.generator = self.physics_generator + self.sigma_generator
-            self.saved_params = {"updatable_params": {"sigma": 0.05},
-                                 "updatable_params_converter": {"sigma": float},
-                                 "fixed_params": {"noise_sigma_min": 0.001, "noise_sigma_max": 0.2,
+            self.physics_generator = GaussianBlurGenerator(psf_size=(psf_size, psf_size),
+                                                           sigma_min=2.0, sigma_max=2.0,
+                                                           num_channels=1,
+                                                           device=device_str) + SigmaGenerator(sigma_min=0.05, sigma_max=0.05, device=device_str)
+            self.generator = self.physics_generator
+            self.saved_params = {"updatable_params": {},
+                                 "updatable_params_converter": {},
+                                 "fixed_params": {"noise_sigma": 0.05, "blur_sigma": 2.0,
+                                                  "psf_size": 31, "num_channels": 1}}
+        elif self.name == "GaussianBlur_hard":
+            psf_size = 31
+            self.physics = dinv.physics.Blur(noise_model=dinv.physics.GaussianNoise(sigma=0.05), padding="valid",
+                                             device=device_str)
+            self.physics_generator = GaussianBlurGenerator(psf_size=(psf_size, psf_size),
+                                                           sigma_min=4.0, sigma_max=4.0,
+                                                           num_channels=1,
+                                                           device=device_str) + SigmaGenerator(sigma_min=0.1, sigma_max=0.1, device=device_str)
+            self.generator = self.physics_generator
+            self.saved_params = {"updatable_params": {},
+                                 "updatable_params_converter": {},
+                                 "fixed_params": {"noise_sigma": 0.1, "blur_sigma": 4.0,
                                                   "psf_size": 31, "num_channels": 1}}
         elif self.name == "MRI":
             self.physics = dinv.physics.MRI(img_size=(640, 320), noise_model=dinv.physics.GaussianNoise(sigma=.01),