Update app.py

app.py

@@ -171,14 +171,19 @@ def image_inference(image):
     # faces = [ image[top:bottom, left:right].copy() for (top, left, bottom, right) in rawfaces ]
     # faces = RetinaFace.extract_faces( image, align = True)
     # faces_mean = [ x.mean() for x in faces ]
-    faces = [ x['face'] for x in  DeepFace.extract_faces( image )]
+    rawfaces = DeepFace.extract_faces( image )
+    faces = [ x['face'] for x in  rawfaces]
     # faces = [ image[x:w+x, y:h+y].copy() for (x, y, w, h) in rawfaces ]
     faces = [ Image.fromarray(x, mode = 'RGB') for x in faces ]
     ages, genders, beards, blurs, ethncities, masks = AnalysisFeatures( faces )
 
     annotatedImage = image.copy()
-    for (top, left, bottom, right) in rawfaces:
-        cv2.rectangle(annotatedImage, (top, left), (left, right), (255, 0, 0), 5)
+    for x in rawfaces:
+        x = x['facial_area']
+        top, left, bottom, right = x['x'], x['y'], x['w'], x['h']
+        bottom += top
+        right += left
+        cv2.rectangle(annotatedImage, (top, bottom), (left, right), (255, 0, 0), 5)
 
     return Image.fromarray(annotatedImage, mode = 'RGB'), {'facesLength':len(faces), 'faceMean':faces_mean }#frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
     # return frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )