From 5ebe3834923f1b5cdaea78f2ff7208201cbc17e5 Mon Sep 17 00:00:00 2001
From: scott-vsi <scott.richardson@visionsystemsinc.com>
Date: Thu, 11 Oct 2018 13:48:39 -0400
Subject: [PATCH] fixed indice

it should be heigh == width, not width == channels
---
 demo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo.py b/demo.py
index ffe0965..c5950c9 100644
--- 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: