mirror of
https://github.com/hanzoai/anomalib.git
synced 2026-10-30 18:48:11 +00:00
zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
25 lines
1.1 KiB
Python
25 lines
1.1 KiB
Python
"""Test if our implementation produces same result as kornia."""
|
|
|
|
# Copyright (C) 2022-2025 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import pytest
|
|
import torch
|
|
from kornia.filters import GaussianBlur2d as korniaGaussianBlur2d
|
|
|
|
from anomalib.models.components import GaussianBlur2d
|
|
|
|
|
|
@pytest.mark.parametrize("sigma", [(4.0, 4.0), (1.9, 3.0), (2.0, 1.5)])
|
|
@pytest.mark.parametrize("channels", list(range(1, 6)))
|
|
@pytest.mark.parametrize("kernel_size", [(33, 33), (9, 9), (11, 5), (3, 3)])
|
|
def test_blur_equivalence(kernel_size: tuple[int, int], sigma: tuple[int, int], channels: int) -> None:
|
|
"""Test if the GaussianBlur2d layer produces the same output as Kornia's GaussianBlur2d layer."""
|
|
for _ in range(10):
|
|
input_tensor = torch.randn((3, channels, 128, 128))
|
|
kornia = korniaGaussianBlur2d(kernel_size, sigma, separable=False)
|
|
blur_kornia = kornia(input_tensor)
|
|
gaussian = GaussianBlur2d(kernel_size=kernel_size, sigma=sigma, channels=channels)
|
|
blur_gaussian = gaussian(input_tensor)
|
|
torch.testing.assert_close(blur_kornia, blur_gaussian)
|