Spaces:

deepinv
/

denoising

Sleeping

Julián Tachella commited on Feb 23, 2024

Commit

c771378

1 Parent(s): 0aa692a

test

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,9 +5,9 @@ import torch
 def image_mod(image):
-    denoiser = dinv.models.BM3D()
     image = torch.tensor(image).unsqueeze(0)
-    image = denoiser(image)
     image = image.squeeze(0).cpu().detach().numpy()
     return image

 def image_mod(image):
+    denoiser = dinv.models.MedianFilter()
     image = torch.tensor(image).unsqueeze(0)
+    image = denoiser(image, 0.1)
     image = image.squeeze(0).cpu().detach().numpy()
     return image