Merge pull request yfeng95#75 from scott-vsi/patch-1

fixed indice
LouisFinner · Oct 18, 2018 · fc12fe5 · fc12fe5
2 parents c2623a6 + 5ebe383
commit fc12fe5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -54,7 +54,7 @@ def main(args):
                 image = (image*255).astype(np.uint8)
             pos = prn.process(image) # use dlib to detect face
         else:
-            if image.shape[1] == image.shape[2]:
+            if image.shape[0] == image.shape[1]:
                 image = resize(image, (256,256))
                 pos = prn.net_forward(image/255.) # input image has been cropped to 256x256
             else: