From 18942f20b71bc552a398fbdc5051b9bd4fe6677f Mon Sep 17 00:00:00 2001 From: Pakkapon Phongthawee Date: Fri, 18 Oct 2019 12:31:40 +0700 Subject: [PATCH] skimage >= 0.16 need to set parameter multichannel skimage >= 0.16 need to set parameter multichannel = True to resize the image if it isn't set it will crash because output will be 1 channel instead of 3 channels. --- demo.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/demo.py b/demo.py index c5950c9..088c956 100644 --- a/demo.py +++ b/demo.py @@ -55,6 +55,10 @@ def main(args): pos = prn.process(image) # use dlib to detect face else: if image.shape[0] == image.shape[1]: + try: + rescale(image, 1000./max_size,multichannel=True) + except: + rescale(image, 1000./max_size) image = resize(image, (256,256)) pos = prn.net_forward(image/255.) # input image has been cropped to 256x256 else: