From ecb204c940bee664bd889240e598eba639a22622 Mon Sep 17 00:00:00 2001 From: shihy Date: Sat, 25 Nov 2023 15:08:13 +0800 Subject: [PATCH] [MNT] Better support for High Definition Images --- learnware/specification/regular/image/rkme.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/learnware/specification/regular/image/rkme.py b/learnware/specification/regular/image/rkme.py index 1f9a4a1..81b68f7 100644 --- a/learnware/specification/regular/image/rkme.py +++ b/learnware/specification/regular/image/rkme.py @@ -130,7 +130,7 @@ class RKMEImageSpecification(RegularStatSpecification): raise ModuleNotFoundError(f"RKMEImageSpecification is not available because 'torchvision' is not installed! Please install it manually." ) if X.shape[2] != RKMEImageSpecification.IMAGE_WIDTH or X.shape[3] != RKMEImageSpecification.IMAGE_WIDTH: - X = Resize((RKMEImageSpecification.IMAGE_WIDTH, RKMEImageSpecification.IMAGE_WIDTH), antialias=None)(X) + X = Resize((RKMEImageSpecification.IMAGE_WIDTH, RKMEImageSpecification.IMAGE_WIDTH), antialias=True)(X) num_points = X.shape[0] X_shape = X.shape