Update avatar.py

avatar.py

@@ -188,7 +188,9 @@ class Avatar:
       predictions = []
       try:
         for i in tqdm(range(0, len(images), self.face_detect_batch_size)):
-          predictions.extend(detector.face_detector.detect_from_batch(np.array(images[i:i + self.face_detect_batch_size])))
+          batch_images = np.array(images[i:i + self.face_detect_batch_size])
+            batch_images = torch.from_numpy(batch_images).permute(0, 3, 1, 2).float().to(self.device)
+            predictions.extend(detector.face_detector.detect_from_batch(batch_images))
       except RuntimeError:
         if self.face_detect_batch_size == 1:
           raise RuntimeError('Image too big to run face detection on GPU. Please use the --resize_factor argument')