jungwoonshin commited on
Commit
6656b25
1 Parent(s): 189275c
Files changed (1) hide show
  1. kernel_utils.py +1 -1
kernel_utils.py CHANGED
@@ -199,7 +199,7 @@ class VideoReader:
199
  class FaceExtractor:
200
  def __init__(self, video_read_fn):
201
  self.video_read_fn = video_read_fn
202
- self.detector = MTCNN(margin=0, thresholds=[0.7, 0.8, 0.8], device="cuda")
203
 
204
  def process_videos(self, video_path):
205
  videos_read = []
 
199
  class FaceExtractor:
200
  def __init__(self, video_read_fn):
201
  self.video_read_fn = video_read_fn
202
+ self.detector = MTCNN(margin=0, thresholds=[0.7, 0.8, 0.8], device="cpu")
203
 
204
  def process_videos(self, video_path):
205
  videos_read = []