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

51
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,36 +110,34 @@ 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 += 1 # increase frame for a single face
face_detected = True
if face_index == 0:
face_included_frames = (
face_included_frames + 1
) # increase frame for a single face
cv2.rectangle(
img, (x, y), (x + w, y + h), (67, 67, 67), 1
) # draw rectangle to main image
cv2.rectangle(
img, (x, y), (x + w, y + h), (67, 67, 67), 1
) # draw rectangle to main image
cv2.putText(
img,
str(frame_threshold - face_included_frames),
(int(x + w / 4), int(y + h / 1.5)),
cv2.FONT_HERSHEY_SIMPLEX,
4,
(255, 255, 255),
2,
)
cv2.putText(
img,
str(frame_threshold - face_included_frames),
(int(x + w / 4), int(y + h / 1.5)),
cv2.FONT_HERSHEY_SIMPLEX,
4,
(255, 255, 255),
2,
)
detected_face = img[int(y) : int(y + h), int(x) : int(x + w)] # crop detected face
detected_face = img[int(y) : int(y + h), int(x) : int(x + w)] # crop detected face
# -------------------------------------
# -------------------------------------
detected_faces.append((x, y, w, h))
face_index = face_index + 1
detected_faces.append((x, y, w, h))
face_index = face_index + 1
# -------------------------------------
# -------------------------------------
if face_detected == True and face_included_frames == frame_threshold and freeze == False:
freeze = True