bit-guber commited on
Commit
a5b7893
1 Parent(s): 46b7408

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -173,17 +173,14 @@ def image_inference(image):
173
  # faces_mean = [ x.mean() for x in faces ]
174
  rawfaces = DeepFace.extract_faces( image )
175
  faces = [ x['face'] for x in rawfaces]
 
176
  # faces = [ image[x:w+x, y:h+y].copy() for (x, y, w, h) in rawfaces ]
177
  faces = [ Image.fromarray(x, mode = 'RGB') for x in faces ]
178
  ages, genders, beards, blurs, ethncities, masks = AnalysisFeatures( faces )
179
 
180
  annotatedImage = image.copy()
181
- for x in rawfaces:
182
- x = x['facial_area']
183
- top, left, bottom, right = x['x'], x['y'], x['w'], x['h']
184
- bottom += top
185
- right += left
186
- cv2.rectangle(annotatedImage, (top, bottom), (left, right), (255, 0, 0), 5)
187
 
188
  return Image.fromarray(annotatedImage, mode = 'RGB'), frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
189
  # return frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
 
173
  # faces_mean = [ x.mean() for x in faces ]
174
  rawfaces = DeepFace.extract_faces( image )
175
  faces = [ x['face'] for x in rawfaces]
176
+ rawfaces = [ (x['facial_area']['x'], x['facial_area']['y'], x['facial_area']['w'], x['facial_area']['h']) for x in rawfaces ]
177
  # faces = [ image[x:w+x, y:h+y].copy() for (x, y, w, h) in rawfaces ]
178
  faces = [ Image.fromarray(x, mode = 'RGB') for x in faces ]
179
  ages, genders, beards, blurs, ethncities, masks = AnalysisFeatures( faces )
180
 
181
  annotatedImage = image.copy()
182
+ for (x, y, w, h) in rawfaces:
183
+ cv2.rectangle(annotatedImage, (x, x+w), (y, y+h), (255, 0, 0), 5)
 
 
 
 
184
 
185
  return Image.fromarray(annotatedImage, mode = 'RGB'), frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
186
  # return frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )