Fix blur in Sobel filter (#389)

* Bug fix in Sobel.

* Update source notebook

---------

Co-authored-by: Benjamin Aubert <baubert@eos-imaging.com>
Co-authored-by: Vivek Gopalakrishnan <vivekg@mit.edu>

Author: 3 people

diffdrr/metrics.py

@@ -67,7 +67,7 @@ def forward(self, x1, x2):
 
 
 class Sobel(torch.nn.Module):
-    def __init__(self, sigma):
+    def __init__(self, sigma: float):
         super().__init__()
         self.sigma = sigma
         self.filter = torch.nn.Conv2d(
@@ -86,7 +86,7 @@ def __init__(self, sigma):
 
     def forward(self, img):
         x = gaussian_blur(img, 5, self.sigma)
-        x = self.filter(img)
+        x = self.filter(x)
         return x
 
 # %% ../notebooks/api/05_metrics.ipynb 10

notebooks/api/05_metrics.ipynb

@@ -161,7 +161,7 @@
     "\n",
     "\n",
     "class Sobel(torch.nn.Module):\n",
-    "    def __init__(self, sigma):\n",
+    "    def __init__(self, sigma: float):\n",
     "        super().__init__()\n",
     "        self.sigma = sigma\n",
     "        self.filter = torch.nn.Conv2d(\n",
@@ -180,7 +180,7 @@
     "\n",
     "    def forward(self, img):\n",
     "        x = gaussian_blur(img, 5, self.sigma)\n",
-    "        x = self.filter(img)\n",
+    "        x = self.filter(x)\n",
     "        return x"
    ]
   },