Spaces:
Sleeping
Sleeping
added inpainting
Browse files- factories.py +6 -5
factories.py
CHANGED
@@ -84,17 +84,18 @@ class PhysicsWithGenerator(torch.nn.Module):
|
|
84 |
"fixed_params": {"noise_sigma_min": 0.05, "noise_sigma_max": 0.05,
|
85 |
"blur_sigma": 4.0, "psf_size": 31, "num_channels": 1}}
|
86 |
elif self.name == "Inpainting":
|
|
|
|
|
|
|
87 |
self.physics = dinv.physics.Inpainting(tensor_size=(256, 256), mask=split_ratio,
|
88 |
noise_model=dinv.physics.GaussianNoise(sigma=sigma),
|
89 |
device=device_str)
|
90 |
self.physics_generator = dinv.physics.generator.BernoulliSplittingMaskGenerator((3, 256, 256),
|
91 |
split_ratio=split_ratio, pixelwise=pixelwise,
|
92 |
-
random_split_ratio=True,
|
93 |
-
max_split_ratio=split_ratio, device=device_str)
|
94 |
self.generator = dinv.physics.generator.BernoulliSplittingMaskGenerator((3, 256, 256),
|
95 |
-
|
96 |
-
|
97 |
-
max_split_ratio=split_ratio, device=device_str)
|
98 |
|
99 |
self.saved_params = {"updatable_params": {},
|
100 |
"updatable_params_converter": {"sigma": float},
|
|
|
84 |
"fixed_params": {"noise_sigma_min": 0.05, "noise_sigma_max": 0.05,
|
85 |
"blur_sigma": 4.0, "psf_size": 31, "num_channels": 1}}
|
86 |
elif self.name == "Inpainting":
|
87 |
+
sigma = 0.05
|
88 |
+
split_ratio = 0.3
|
89 |
+
pixelwise = False
|
90 |
self.physics = dinv.physics.Inpainting(tensor_size=(256, 256), mask=split_ratio,
|
91 |
noise_model=dinv.physics.GaussianNoise(sigma=sigma),
|
92 |
device=device_str)
|
93 |
self.physics_generator = dinv.physics.generator.BernoulliSplittingMaskGenerator((3, 256, 256),
|
94 |
split_ratio=split_ratio, pixelwise=pixelwise,
|
95 |
+
random_split_ratio=True, device=device_str)
|
|
|
96 |
self.generator = dinv.physics.generator.BernoulliSplittingMaskGenerator((3, 256, 256),
|
97 |
+
split_ratio=split_ratio, pixelwise=pixelwise,
|
98 |
+
random_split_ratio=True, device=device_str)
|
|
|
99 |
|
100 |
self.saved_params = {"updatable_params": {},
|
101 |
"updatable_params_converter": {"sigma": float},
|