bit-guber commited on
Commit
831787b
1 Parent(s): f3c3925

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -157,8 +157,9 @@ def image_inference(image):
157
  # Draw rectangle around the faces
158
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
159
 
160
- rawfaces = face_recognition.face_locations( image, number_of_times_to_upsample = 5 , model="hog")
161
  faces = [ image[top:bottom, left:right].copy() for (top, left, bottom, right) in rawfaces ]
 
162
  # faces = [ image[x:w+x, y:h+y].copy() for (x, y, w, h) in rawfaces ]
163
  faces = [ Image.fromarray(x, mode = 'RGB') for x in faces ]
164
  ages, genders, beards, blurs, ethncities, masks = AnalysisFeatures( faces )
@@ -167,7 +168,7 @@ def image_inference(image):
167
  for (top, left, bottom, right) in rawfaces:
168
  cv2.rectangle(annotatedImage, (top, left), (left, right), (255, 0, 0), 5)
169
 
170
- return Image.fromarray(annotatedImage, mode = 'RGB'), frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
171
  # return frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
172
  def video_inference(video_path):
173
 
@@ -191,7 +192,7 @@ def video_inference(video_path):
191
  except:
192
  break
193
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
194
- rawfaces = face_recognition.face_locations( image, number_of_times_to_upsample = 5 , model="hog")
195
  # Draw rectangle around the faces
196
  # https://stackoverflow.com/questions/15589517/how-to-crop-an-image-in-opencv-using-python for fliping axis
197
  global_facesCo.append( rawfaces )
 
157
  # Draw rectangle around the faces
158
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
159
 
160
+ rawfaces = face_recognition.face_locations( image, number_of_times_to_upsample = 1 , model="hog")
161
  faces = [ image[top:bottom, left:right].copy() for (top, left, bottom, right) in rawfaces ]
162
+ faces_mean = [ x.mean() for x in faces ]
163
  # faces = [ image[x:w+x, y:h+y].copy() for (x, y, w, h) in rawfaces ]
164
  faces = [ Image.fromarray(x, mode = 'RGB') for x in faces ]
165
  ages, genders, beards, blurs, ethncities, masks = AnalysisFeatures( faces )
 
168
  for (top, left, bottom, right) in rawfaces:
169
  cv2.rectangle(annotatedImage, (top, left), (left, right), (255, 0, 0), 5)
170
 
171
+ return Image.fromarray(annotatedImage, mode = 'RGB'), {'facesLength':len(faces), 'faceMean':faces_mean }#frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
172
  # return frameWrapper( rawfaces, ages, genders, beards, blurs, ethncities, masks )
173
  def video_inference(video_path):
174
 
 
192
  except:
193
  break
194
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
195
+ rawfaces = face_recognition.face_locations( image, number_of_times_to_upsample = 1 , model="hog")
196
  # Draw rectangle around the faces
197
  # https://stackoverflow.com/questions/15589517/how-to-crop-an-image-in-opencv-using-python for fliping axis
198
  global_facesCo.append( rawfaces )