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: