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

video path is enabled in stream

Browse Source
This commit is contained in:
Mehrab Shahbazi 2025-01-01 12:29:20 +03:30
parent ca6afd98ef
commit 33a502f609
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
deepface/modules/streaming.py
View File

@ -22,6 +22,7 @@ os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
IDENTIFIED_IMG_SIZE = 112
TEXT_COLOR = (255, 255, 255)
# pylint: disable=unused-variable
def analysis(
db_path: str,
@ -82,7 +83,11 @@ def analysis(
num_frames_with_faces = 0
tic = time.time()
cap = cv2.VideoCapture(source) # webcam
# If source is an integer, use it as a webcam index. Otherwise, treat it as a video file path.
if isinstance(source, int):
cap = cv2.VideoCapture(source) # webcam
else:
cap = cv2.VideoCapture(str(source)) # video file
while True:
has_frame, img = cap.read()
if not has_frame: