From 96931d6619d0b8456c89250fb6912b2664ba136c Mon Sep 17 00:00:00 2001 From: iperov Date: Sat, 7 Aug 2021 00:28:52 +0400 Subject: [PATCH] fix ImageProcessor.py --- xlib/image/ImageProcessor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/xlib/image/ImageProcessor.py b/xlib/image/ImageProcessor.py index 5997794..97cd9ea 100644 --- a/xlib/image/ImageProcessor.py +++ b/xlib/image/ImageProcessor.py @@ -152,7 +152,7 @@ class ImageProcessor: else: img = sp.ndimage.zoom(img, (scale, scale, 1.0), order=1) - H,W,_ = img.shape + H,W = img.shape[0:2] img = img.reshape( (H,W,N,C) ).transpose( (2,0,1,3) ) if pad_to_target: