MTCNN CPU

kernel_utils.py

@@ -199,7 +199,7 @@ class VideoReader:
 class FaceExtractor:
     def __init__(self, video_read_fn):
         self.video_read_fn = video_read_fn
-        self.detector = MTCNN(margin=0, thresholds=[0.7, 0.8, 0.8], device="cuda")
+        self.detector = MTCNN(margin=0, thresholds=[0.7, 0.8, 0.8], device="cpu")
 
     def process_videos(self, input_dir, filenames, video_idxs):
         videos_read = []