jarondon82 commited on
Commit
3911742
·
1 Parent(s): 344e1a0

Enable eye and smile detection in video processing

Browse files
Files changed (1) hide show
  1. streamlit_app.py +5 -5
streamlit_app.py CHANGED
@@ -426,7 +426,7 @@ def main():
426
  return href
427
 
428
  # Function to process video frames
429
- def process_video(video_path, face_net, eye_cascade, smile_cascade, conf_threshold=0.5, detect_eyes=False, detect_smile=False, bbox_color=(0, 255, 0), smile_sensitivity=15, eye_sensitivity=5):
430
  cap = cv2.VideoCapture(video_path)
431
 
432
  # Get video properties
@@ -685,8 +685,8 @@ def main():
685
  eye_cascade,
686
  smile_cascade,
687
  conf_threshold,
688
- detect_eyes=False,
689
- detect_smile=False,
690
  bbox_color=bbox_color_bgr,
691
  eye_sensitivity=5
692
  )
@@ -940,8 +940,8 @@ def main():
940
  eye_cascade,
941
  smile_cascade,
942
  conf_threshold,
943
- detect_eyes=detect_eyes,
944
- detect_smile=detect_smile,
945
  bbox_color=bbox_color_bgr,
946
  smile_sensitivity=smile_sensitivity,
947
  eye_sensitivity=eye_sensitivity
 
426
  return href
427
 
428
  # Function to process video frames
429
+ def process_video(video_path, face_net, eye_cascade, smile_cascade, conf_threshold=0.5, detect_eyes=True, detect_smile=True, bbox_color=(0, 255, 0), smile_sensitivity=15, eye_sensitivity=5):
430
  cap = cv2.VideoCapture(video_path)
431
 
432
  # Get video properties
 
685
  eye_cascade,
686
  smile_cascade,
687
  conf_threshold,
688
+ detect_eyes=True,
689
+ detect_smile=True,
690
  bbox_color=bbox_color_bgr,
691
  eye_sensitivity=5
692
  )
 
940
  eye_cascade,
941
  smile_cascade,
942
  conf_threshold,
943
+ detect_eyes=True,
944
+ detect_smile=True,
945
  bbox_color=bbox_color_bgr,
946
  smile_sensitivity=smile_sensitivity,
947
  eye_sensitivity=eye_sensitivity