bit-guber commited on
Commit
54f30b7
1 Parent(s): e2d5be3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -148,7 +148,7 @@ def image_inference(image):
148
 
149
 
150
  if sum(image.shape) == 0:
151
- return { 'ErrorFound': 'ImageNotFound' }
152
  # Convert into grayscale
153
  # gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
154
  # Detect faces
@@ -157,7 +157,7 @@ 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 , model="cnn")
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 ]
@@ -191,7 +191,7 @@ def video_inference(video_path):
191
  except:
192
  break
193
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
194
- rawfaces = face_recognition.face_locations( image , model="cnn")
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 )
 
148
 
149
 
150
  if sum(image.shape) == 0:
151
+ return image, { 'ErrorFound': 'ImageNotFound' }
152
  # Convert into grayscale
153
  # gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
154
  # Detect faces
 
157
  # Draw rectangle around the faces
158
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
159
 
160
+ rawfaces = face_recognition.face_locations( image , 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 ]
 
191
  except:
192
  break
193
  # rawfaces = postProcessed( rawfaces, image.shape[:2] )
194
+ rawfaces = face_recognition.face_locations( image , 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 )