From 19f6e51d450e17b116b62976233b2c07d6d79c10 Mon Sep 17 00:00:00 2001 From: Eugene Khvedchenya Date: Fri, 9 Feb 2024 16:45:40 +0200 Subject: [PATCH] Added explicit antialias=False to ensure we can export this operation to ONNX --- tests/unit_tests/export_onnx_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit_tests/export_onnx_test.py b/tests/unit_tests/export_onnx_test.py index cc4dba3cdb..3973329439 100644 --- a/tests/unit_tests/export_onnx_test.py +++ b/tests/unit_tests/export_onnx_test.py @@ -12,7 +12,7 @@ class TestModelsONNXExport(unittest.TestCase): def test_models_onnx_export_with_deprecated_input_shape(self): pretrained_model = models.get(Models.RESNET18, num_classes=1000, pretrained_weights="imagenet") - preprocess = Compose([Resize(224), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) + preprocess = Compose([Resize(224, antialias=False), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) with tempfile.TemporaryDirectory() as tmpdirname: out_path = os.path.join(tmpdirname, "resnet18.onnx") models.convert_to_onnx(model=pretrained_model, out_path=out_path, input_shape=(3, 256, 256), pre_process=preprocess) @@ -20,7 +20,7 @@ def test_models_onnx_export_with_deprecated_input_shape(self): def test_models_onnx_export(self): pretrained_model = models.get(Models.RESNET18, num_classes=1000, pretrained_weights="imagenet") - preprocess = Compose([Resize(224), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) + preprocess = Compose([Resize(224, antialias=False), Standardize(), Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])]) with tempfile.TemporaryDirectory() as tmpdirname: out_path = os.path.join(tmpdirname, "resnet18.onnx") models.convert_to_onnx(