🐞 Fix dimensionality mismatch issue caused by the new kornia version (#1944)

* Address dimension issue caused by different kornia versions

Signed-off-by: Samet Akcay <[email protected]>

* Unbound the kornia version in pyproject.toml file

Signed-off-by: Samet Akcay <[email protected]>

* Update src/anomalib/models/components/filters/blur.py

Co-authored-by: Dick Ameln <[email protected]>

---------

Signed-off-by: Samet Akcay <[email protected]>
Co-authored-by: Dick Ameln <[email protected]>

Author: samet-akcay

pyproject.toml

@@ -29,7 +29,7 @@ core = [
     "einops>=0.3.2",
     "freia>=0.2",
     "imgaug==0.4.0",
-    "kornia>=0.6.6,<0.6.10",
+    "kornia>=0.6.6",
     "matplotlib>=3.4.3",
     "opencv-python>=4.5.3.56",
     "pandas>=1.1.0",

src/anomalib/models/components/filters/blur.py

@@ -64,7 +64,9 @@ def __init__(
         self.register_buffer("kernel", get_gaussian_kernel2d(kernel_size=kernel_size, sigma=sigma))
         if normalize:
             self.kernel = normalize_kernel2d(self.kernel)
-        self.kernel.unsqueeze_(0).unsqueeze_(0)
+
+        self.kernel = self.kernel.view(1, 1, *self.kernel.shape[-2:])
+
         self.kernel = self.kernel.expand(self.channels, -1, -1, -1)
         self.border_type = border_type
         self.padding = padding