diff --git a/tensorlayer/prepro.py b/tensorlayer/prepro.py index 0d3e2938b..a59e628c6 100644 --- a/tensorlayer/prepro.py +++ b/tensorlayer/prepro.py @@ -1852,11 +1852,9 @@ def imresize(x, size=None, interp='bicubic', mode=None): # greyscale x = scipy.misc.imresize(x[:, :, 0], size, interp=interp, mode=mode) return x[:, :, np.newaxis] - elif x.shape[-1] == 3: - # rgb, bgr .. - return scipy.misc.imresize(x, size, interp=interp, mode=mode) else: - raise Exception("Unsupported channel %d" % x.shape[-1]) + # rgb, bgr, rgba + return scipy.misc.imresize(x, size, interp=interp, mode=mode) # value scale