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},
|