reab5555 commited on
Commit
f88947f
·
verified ·
1 Parent(s): 91ab9c7

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +1 -1
video_processing.py CHANGED
@@ -69,7 +69,7 @@ def process_frames(frames_folder, faces_folder, frame_count, progress):
69
  face_rgb = cv2.cvtColor(face, cv2.COLOR_BGR2RGB)
70
  face_resized = cv2.resize(face_rgb, (160, 160))
71
  output_path = os.path.join(faces_folder, f"frame_{frame_num}_face.jpg")
72
- cv2.imwrite(output_path, face_resized_resized)
73
  face_paths.append(output_path)
74
  embedding = get_face_embedding(face_resized)
75
  embeddings_by_frame[frame_num] = embedding
 
69
  face_rgb = cv2.cvtColor(face, cv2.COLOR_BGR2RGB)
70
  face_resized = cv2.resize(face_rgb, (160, 160))
71
  output_path = os.path.join(faces_folder, f"frame_{frame_num}_face.jpg")
72
+ cv2.imwrite(output_path, face_resized)
73
  face_paths.append(output_path)
74
  embedding = get_face_embedding(face_resized)
75
  embeddings_by_frame[frame_num] = embedding