Skip to content

Commit b5536b5

Browse files
authored
Update ldm_super_resolution_evaluator.py
1 parent 6c50a39 commit b5536b5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,7 @@ def preprocess_image(image):
261261
@staticmethod
262262
def postprocess_image(image: np.ndarray, std=255, mean=0):
263263
image = image / 2 + 0.5
264+
image = image.transpose(0, 2, 3, 1)
264265
image *= np.array(std, dtype=image.dtype)
265266
image += np.array(mean, dtype=image.dtype)
266267

0 commit comments

Comments
 (0)