Update kernel_utils.py

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 = []
@@ -322,10 +322,10 @@ def predict_on_video(face_extractor, video_path, batch_size, input_size, models,
                     else:
                         pass
             if n > 0:
-                x = torch.tensor(x, device="cuda").float()
+                x = torch.tensor(x, device="cpu").float()
                 # Preprocess the images.
                 x = x.permute((0, 3, 1, 2))
-                # x = x.to('cpu')
+                x = x.to('cpu')
                 for i in range(len(x)):
                     x[i] = normalize_transform(x[i] / 255.)
                 # Make a prediction, then take the average.