github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-08 04:25:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use return boolean to detect effective capture

Browse Source 
Actually an image is always returned but empty
This commit is contained in:
Andrea Lanfranchi 2024-02-21 15:56:47 +01:00
parent 810dfac4fe
commit 051a8d3172
No known key found for this signature in database
GPG Key ID: 2D623A60BB279915
1 changed files with 24 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
deepface/modules/realtime.py
View File

@ -68,9 +68,8 @@ def analysis(
cap = cv2.VideoCapture(source) # webcam
while True:
_, img = cap.read()
if img is None:
has_frame, img = cap.read()
if not has_frame:
break
# cv2.namedWindow('img', cv2.WINDOW_FREERATIO)
@ -111,13 +110,11 @@ def analysis(
detected_faces = []
face_index = 0
for x, y, w, h in faces:
if w > 130: # discard small detected faces
if w <= 130: # discard small detected faces
continue
face_detected = True
if face_index == 0:
face_included_frames = (
face_included_frames + 1
) # increase frame for a single face
face_included_frames += 1 # increase frame for a single face
cv2.rectangle(
img, (x, y), (x + w, y + h), (67, 67, 67), 1