ZF/uniface
1
0
Fork 0
mirror of https://github.com/yakhyo/uniface.git synced 2025-12-30 09:02:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Code style formatting changes

Browse Source
This commit is contained in:
yakhyo
2025-11-26 00:05:24 +09:00
parent 0771a7959a
commit 84cda5f56c
12 changed files with 172 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
scripts/run_video_detection.py
View File

@@ -11,7 +11,13 @@ from uniface import SCRFD, RetinaFace
from uniface.visualization import draw_detections
def process_video(detector, input_path: str, output_path: str, threshold: float = 0.6, show_preview: bool = False):
def process_video(
detector,
input_path: str,
output_path: str,
threshold: float = 0.6,
show_preview: bool = False,
):
cap = cv2.VideoCapture(input_path)
if not cap.isOpened():
print(f"Error: Cannot open video file '{input_path}'")
@@ -51,7 +57,15 @@ def process_video(detector, input_path: str, output_path: str, threshold: float
landmarks = [f["landmarks"] for f in faces]
draw_detections(frame, bboxes, scores, landmarks, vis_threshold=threshold)
cv2.putText(frame, f"Faces: {len(faces)}", (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
cv2.putText(
frame,
f"Faces: {len(faces)}",
(10, 30),
cv2.FONT_HERSHEY_SIMPLEX,
1,
(0, 255, 0),
2,
)
out.write(frame)
if show_preview: