sdafd commited on
Commit
b76f2d8
1 Parent(s): 4aa3d02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from mtcnn.mtcnn import MTCNN
3
  import cv2
4
  import numpy as np
 
5
 
6
  # Function to detect faces using MTCNN
7
  def detect_faces(image):
@@ -12,12 +13,15 @@ def detect_faces(image):
12
  detector = MTCNN()
13
  faces = detector.detect_faces(image_rgb)
14
 
 
 
 
15
  # Draw bounding boxes around detected faces
16
  for face in faces:
17
  x, y, width, height = face['box']
18
  cv2.rectangle(image, (x, y), (x + width, y + height), (255, 0, 0), 2)
19
 
20
- return image, faces
21
 
22
  # Gradio Interface
23
  iface = gr.Interface(
 
2
  from mtcnn.mtcnn import MTCNN
3
  import cv2
4
  import numpy as np
5
+ import json
6
 
7
  # Function to detect faces using MTCNN
8
  def detect_faces(image):
 
13
  detector = MTCNN()
14
  faces = detector.detect_faces(image_rgb)
15
 
16
+ # Extract and format face information
17
+ formatted_faces = [{"x": face['box'][0], "y": face['box'][1], "width": face['box'][2], "height": face['box'][3]} for face in faces]
18
+
19
  # Draw bounding boxes around detected faces
20
  for face in faces:
21
  x, y, width, height = face['box']
22
  cv2.rectangle(image, (x, y), (x + width, y + height), (255, 0, 0), 2)
23
 
24
+ return image, json.dumps(formatted_faces)
25
 
26
  # Gradio Interface
27
  iface = gr.Interface(