mterris commited on
Commit
a49aa2c
·
1 Parent(s): ee6f900

added inpainting

Browse files
Files changed (1) hide show
  1. 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, min_split_ratio=split_ratio,
93
- max_split_ratio=split_ratio, device=device_str)
94
  self.generator = dinv.physics.generator.BernoulliSplittingMaskGenerator((3, 256, 256),
95
- split_ratio=split_ratio, pixelwise=pixelwise,
96
- random_split_ratio=True, min_split_ratio=split_ratio,
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},