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)