Browse Source

Merge branch 'main' of https://github.com/Learnware-LAMDA/Learnware into search_result

tags/v0.3.2
bxdd 2 years ago
parent
commit
eb9ab56f1c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      learnware/specification/regular/image/rkme.py

+ 1
- 1
learnware/specification/regular/image/rkme.py View File

@@ -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


Loading…
Cancel
Save