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

ensure output path

Browse Source
This commit is contained in:
Mehrab Shahbazi 2025-01-03 14:55:14 +03:30
parent aa47e33cf4
commit 5af32fa841
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
deepface/modules/streaming.py
View File

@ -91,7 +91,9 @@ def analysis(
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS) fps = cap.get(cv2.CAP_PROP_FPS)
fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for output file fourcc = cv2.VideoWriter_fourcc(*"mp4v") # Codec for output file
# Ensure the output directory exists if output_path is provided
if output_path:
os.makedirs(os.path.dirname(output_path), exist_ok=True)
# Initialize video writer if output_path is provided # Initialize video writer if output_path is provided
video_writer = ( video_writer = (
cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (width, height)) cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*"mp4v"), fps, (width, height))
@ -254,10 +256,10 @@ def search_identity(
# detected face is coming from parent, safe to access 1st index # detected face is coming from parent, safe to access 1st index
df = dfs[0] df = dfs[0]
if df.shape[0] == 0: if df.shape[0] == 0:
return None, None return None, None
candidate = df.iloc[0] candidate = df.iloc[0]
target_path = candidate["identity"] target_path = candidate["identity"]
logger.info(f"Hello, {target_path}") logger.info(f"Hello, {target_path}")