reab5555 commited on
Commit
3e62a2f
·
verified ·
1 Parent(s): a4c7349

Update video_processing.py

Browse files
Files changed (1) hide show
  1. video_processing.py +1 -1
video_processing.py CHANGED
@@ -18,7 +18,7 @@ import torch
18
  import mediapipe as mp
19
 
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
- mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.95, 0.95, 0.95], min_face_size=150)
22
 
23
  def extract_frames(video_path, output_folder, desired_fps, progress_callback=None):
24
  os.makedirs(output_folder, exist_ok=True)
 
18
  import mediapipe as mp
19
 
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21
+ mtcnn = MTCNN(keep_all=False, device=device, thresholds=[0.95, 0.95, 0.95], min_face_size=200)
22
 
23
  def extract_frames(video_path, output_folder, desired_fps, progress_callback=None):
24
  os.makedirs(output_folder, exist_ok=True)