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.
74 lines
2.7 KiB
Python
74 lines
2.7 KiB
Python
"""Test the pre-processing transforms utils."""
|
|
|
|
# Copyright (C) 2024 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from torchvision.transforms.v2 import CenterCrop, Compose, Resize, ToTensor
|
|
|
|
from anomalib.data.transforms import ExportableCenterCrop
|
|
from anomalib.pre_processing.utils.transform import (
|
|
convert_center_crop_transform,
|
|
disable_antialiasing,
|
|
get_exportable_transform,
|
|
)
|
|
|
|
|
|
def test_get_exportable_transform() -> None:
|
|
"""Test the get_exportable_transform function."""
|
|
# Test with None transform
|
|
assert get_exportable_transform(None) is None
|
|
|
|
# Test with Resize transform
|
|
resize = Resize((224, 224), antialias=True)
|
|
exportable_resize = get_exportable_transform(resize)
|
|
assert isinstance(exportable_resize, Resize)
|
|
assert not exportable_resize.antialias
|
|
|
|
# Test with CenterCrop transform
|
|
center_crop = CenterCrop((224, 224))
|
|
exportable_center_crop = get_exportable_transform(center_crop)
|
|
assert isinstance(exportable_center_crop, ExportableCenterCrop)
|
|
|
|
# Test with Compose transform
|
|
compose = Compose([Resize((224, 224), antialias=True), CenterCrop((200, 200))])
|
|
exportable_compose = get_exportable_transform(compose)
|
|
assert isinstance(exportable_compose, Compose)
|
|
assert isinstance(exportable_compose.transforms[0], Resize)
|
|
assert not exportable_compose.transforms[0].antialias
|
|
assert isinstance(exportable_compose.transforms[1], ExportableCenterCrop)
|
|
|
|
|
|
def test_disable_antialiasing() -> None:
|
|
"""Test the disable_antialiasing function."""
|
|
# Test with Resize transform
|
|
resize = Resize((224, 224), antialias=True)
|
|
disabled_resize = disable_antialiasing(resize)
|
|
assert not disabled_resize.antialias
|
|
|
|
# Test with Compose transform
|
|
compose = Compose([Resize((224, 224), antialias=True), ToTensor()])
|
|
disabled_compose = disable_antialiasing(compose)
|
|
assert not disabled_compose.transforms[0].antialias
|
|
|
|
# Test with non-Resize transform
|
|
to_tensor = ToTensor()
|
|
assert disable_antialiasing(to_tensor) == to_tensor
|
|
|
|
|
|
def test_convert_centercrop() -> None:
|
|
"""Test the convert_centercrop function."""
|
|
# Test with CenterCrop transform
|
|
center_crop = CenterCrop((224, 224))
|
|
converted_crop = convert_center_crop_transform(center_crop)
|
|
assert isinstance(converted_crop, ExportableCenterCrop)
|
|
assert converted_crop.size == list(center_crop.size)
|
|
|
|
# Test with Compose transform
|
|
compose = Compose([Resize((256, 256)), CenterCrop((224, 224))])
|
|
converted_compose = convert_center_crop_transform(compose)
|
|
assert isinstance(converted_compose.transforms[1], ExportableCenterCrop)
|
|
|
|
# Test with non-CenterCrop transform
|
|
resize = Resize((224, 224))
|
|
assert convert_center_crop_transform(resize) == resize
|