Update ldm_super_resolution_evaluator.py

openvinotoolkit · Oct 29, 2024 · b5536b5 · b5536b5
1 parent 6c50a39
commit b5536b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...y_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py b/...y_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)