From b5536b5a8d99d1e776b9cf6f501df4a404df69de Mon Sep 17 00:00:00 2001 From: Piotr Wolnowski Date: Tue, 29 Oct 2024 16:53:43 +0100 Subject: [PATCH] Update ldm_super_resolution_evaluator.py --- .../custom_evaluators/ldm_super_resolution_evaluator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py b/tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py index 377be4eb58..0855ad1464 100644 --- a/tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py +++ b/tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py @@ -261,6 +261,7 @@ def preprocess_image(image): @staticmethod def postprocess_image(image: np.ndarray, std=255, mean=0): image = image / 2 + 0.5 + image = image.transpose(0, 2, 3, 1) image *= np.array(std, dtype=image.dtype) image += np.array(mean, dtype=image.dtype)