fix input format error

TreB1eN · Nov 2, 2018 · 1d9299d · 1d9299d
1 parent c896f19
commit 1d9299d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/face_verify.py b/face_verify.py
@@ -50,7 +50,8 @@
         isSuccess,frame = cap.read()
         if isSuccess:            
             try:
-                image = Image.fromarray(frame[...,::-1]) #bgr to rgb
+#                 image = Image.fromarray(frame[...,::-1]) #bgr to rgb
+                image = Image.fromarray(frame)
                 bboxes, faces = mtcnn.align_multi(image, conf.face_limit, conf.min_face_size)
                 bboxes = bboxes[:,:-1] #shape:[10,4],only keep 10 highest possibiity faces
                 bboxes = bboxes.astype(int)

diff --git a/infer_on_video.py b/infer_on_video.py
@@ -56,7 +56,8 @@
     while cap.isOpened():
         isSuccess,frame = cap.read()
         if isSuccess:            
-            image = Image.fromarray(frame[...,::-1]) #bgr to rgb
+#             image = Image.fromarray(frame[...,::-1]) #bgr to rgb
+            image = Image.fromarray(frame)
             try:
                 bboxes, faces = mtcnn.align_multi(image, conf.face_limit, 16)
             except: