voontaojizen commited on
Commit
bd67868
·
1 Parent(s): 133f505
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. app.py +7 -6
  2. registered_images/Grayson/image_1.png +0 -0
  3. registered_images/Grayson/image_10.png +0 -0
  4. registered_images/Grayson/image_11.png +0 -0
  5. registered_images/Grayson/image_12.png +0 -0
  6. registered_images/Grayson/image_13.png +0 -0
  7. registered_images/Grayson/image_14.png +0 -0
  8. registered_images/Grayson/image_15.png +0 -0
  9. registered_images/Grayson/image_16.png +0 -0
  10. registered_images/Grayson/image_17.png +0 -0
  11. registered_images/Grayson/image_18.png +0 -0
  12. registered_images/Grayson/image_19.png +0 -0
  13. registered_images/Grayson/image_2.png +0 -0
  14. registered_images/Grayson/image_20.png +0 -0
  15. registered_images/Grayson/image_21.png +0 -0
  16. registered_images/Grayson/image_22.png +0 -0
  17. registered_images/Grayson/image_23.png +0 -0
  18. registered_images/Grayson/image_24.png +0 -0
  19. registered_images/Grayson/image_25.png +0 -0
  20. registered_images/Grayson/image_26.png +0 -0
  21. registered_images/Grayson/image_27.png +0 -0
  22. registered_images/Grayson/image_28.png +0 -0
  23. registered_images/Grayson/image_29.png +0 -0
  24. registered_images/Grayson/image_3.png +0 -0
  25. registered_images/Grayson/image_30.png +0 -0
  26. registered_images/Grayson/image_4.png +0 -0
  27. registered_images/Grayson/image_5.png +0 -0
  28. registered_images/Grayson/image_6.png +0 -0
  29. registered_images/Grayson/image_7.png +0 -0
  30. registered_images/Grayson/image_8.png +0 -0
  31. registered_images/Grayson/image_9.png +0 -0
  32. registered_images/Jentz/image_1.png +0 -0
  33. registered_images/Jentz/image_2.png +0 -0
  34. registered_images/Jentz/image_3.png +0 -0
  35. registered_images/Jerad/image_1.png +0 -0
  36. registered_images/Jerad/image_10.png +0 -0
  37. registered_images/Jerad/image_11.png +0 -0
  38. registered_images/Jerad/image_12.png +0 -0
  39. registered_images/Jerad/image_13.png +0 -0
  40. registered_images/Jerad/image_14.png +0 -0
  41. registered_images/Jerad/image_15.png +0 -0
  42. registered_images/Jerad/image_16.png +0 -0
  43. registered_images/Jerad/image_17.png +0 -0
  44. registered_images/Jerad/image_18.png +0 -0
  45. registered_images/Jerad/image_19.png +0 -0
  46. registered_images/Jerad/image_2.png +0 -0
  47. registered_images/Jerad/image_20.png +0 -0
  48. registered_images/Jerad/image_21.png +0 -0
  49. registered_images/Jerad/image_22.png +0 -0
  50. registered_images/Jerad/image_23.png +0 -0
app.py CHANGED
@@ -14,8 +14,8 @@ from tensorflow.keras.applications.mobilenet_v2 import preprocess_input
14
  MODEL_DIR = 'models/siamese_model-final'
15
  VERIF_IMGS_DIR = 'registered_images'
16
  INPUT_IMG_DIR = 'input_images/cropped_face.png'
17
- THRESHOLD = 0.2
18
- VERIFICATION_THRESHOLD = 0.3
19
  GREEN_COLOR = (0, 255, 0)
20
  RED_COLOR = (0, 0, 255)
21
 
@@ -166,8 +166,8 @@ def detect_faces(image):
166
 
167
  for (x, y, w, h) in faces:
168
  # cv2.rectangle(image, (x, y), (x + w, y + h), GREEN_COLOR, 2)
169
- cropped_image = crop_img(image, (x, y, w, h))
170
- is_real = is_real_face(cropped_image)
171
  if is_real:
172
  cv2.putText(image, "Real", (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, GREEN_COLOR, 2)
173
  cv2.rectangle(image, (x, y), (x + w, y + h), GREEN_COLOR, 2)
@@ -253,10 +253,11 @@ with col2:
253
 
254
  if image_file and len(detected_faces) > 0:
255
 
256
- cropped_image = crop_img(image_np, detected_faces[0])
257
 
258
  temp_image_path = "cropped_face.jpg"
259
- cv2.imwrite(temp_image_path, cv2.cvtColor(cropped_image, cv2.COLOR_RGB2BGR))
 
260
  INPUT_IMG_DIR = temp_image_path
261
 
262
  with st.spinner('Processing...'):
 
14
  MODEL_DIR = 'models/siamese_model-final'
15
  VERIF_IMGS_DIR = 'registered_images'
16
  INPUT_IMG_DIR = 'input_images/cropped_face.png'
17
+ THRESHOLD = 0.9
18
+ VERIFICATION_THRESHOLD = 0.93
19
  GREEN_COLOR = (0, 255, 0)
20
  RED_COLOR = (0, 0, 255)
21
 
 
166
 
167
  for (x, y, w, h) in faces:
168
  # cv2.rectangle(image, (x, y), (x + w, y + h), GREEN_COLOR, 2)
169
+ # cropped_image = crop_img(image, (x, y, w, h))
170
+ is_real = is_real_face(image)
171
  if is_real:
172
  cv2.putText(image, "Real", (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, GREEN_COLOR, 2)
173
  cv2.rectangle(image, (x, y), (x + w, y + h), GREEN_COLOR, 2)
 
253
 
254
  if image_file and len(detected_faces) > 0:
255
 
256
+ # cropped_image = crop_img(image_np, detected_faces[0])
257
 
258
  temp_image_path = "cropped_face.jpg"
259
+ # cv2.imwrite(temp_image_path, cv2.cvtColor(cropped_image, cv2.COLOR_RGB2BGR))
260
+ cv2.imwrite(temp_image_path, cv2.cvtColor(image_np, cv2.COLOR_RGB2BGR))
261
  INPUT_IMG_DIR = temp_image_path
262
 
263
  with st.spinner('Processing...'):
registered_images/Grayson/image_1.png CHANGED
registered_images/Grayson/image_10.png ADDED
registered_images/Grayson/image_11.png ADDED
registered_images/Grayson/image_12.png ADDED
registered_images/Grayson/image_13.png ADDED
registered_images/Grayson/image_14.png ADDED
registered_images/Grayson/image_15.png ADDED
registered_images/Grayson/image_16.png ADDED
registered_images/Grayson/image_17.png ADDED
registered_images/Grayson/image_18.png ADDED
registered_images/Grayson/image_19.png ADDED
registered_images/Grayson/image_2.png CHANGED
registered_images/Grayson/image_20.png ADDED
registered_images/Grayson/image_21.png ADDED
registered_images/Grayson/image_22.png ADDED
registered_images/Grayson/image_23.png ADDED
registered_images/Grayson/image_24.png ADDED
registered_images/Grayson/image_25.png ADDED
registered_images/Grayson/image_26.png ADDED
registered_images/Grayson/image_27.png ADDED
registered_images/Grayson/image_28.png ADDED
registered_images/Grayson/image_29.png ADDED
registered_images/Grayson/image_3.png CHANGED
registered_images/Grayson/image_30.png ADDED
registered_images/Grayson/image_4.png ADDED
registered_images/Grayson/image_5.png ADDED
registered_images/Grayson/image_6.png ADDED
registered_images/Grayson/image_7.png ADDED
registered_images/Grayson/image_8.png ADDED
registered_images/Grayson/image_9.png ADDED
registered_images/Jentz/image_1.png DELETED
Binary file (56.9 kB)
 
registered_images/Jentz/image_2.png DELETED
Binary file (56.8 kB)
 
registered_images/Jentz/image_3.png DELETED
Binary file (57.1 kB)
 
registered_images/Jerad/image_1.png ADDED
registered_images/Jerad/image_10.png ADDED
registered_images/Jerad/image_11.png ADDED
registered_images/Jerad/image_12.png ADDED
registered_images/Jerad/image_13.png ADDED
registered_images/Jerad/image_14.png ADDED
registered_images/Jerad/image_15.png ADDED
registered_images/Jerad/image_16.png ADDED
registered_images/Jerad/image_17.png ADDED
registered_images/Jerad/image_18.png ADDED
registered_images/Jerad/image_19.png ADDED
registered_images/Jerad/image_2.png ADDED
registered_images/Jerad/image_20.png ADDED
registered_images/Jerad/image_21.png ADDED
registered_images/Jerad/image_22.png ADDED
registered_images/Jerad/image_23.png ADDED