Spaces:
Build error
Build error
Commit
·
3911742
1
Parent(s):
344e1a0
Enable eye and smile detection in video processing
Browse files- 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=
|
| 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=
|
| 689 |
-
detect_smile=
|
| 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=
|
| 944 |
-
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
|