numpy throwing error in batch mode when some images do not contain a face about facenet-pytorch HOT 5 OPEN

# Detect faces
batch_boxes, batch_probs, batch_points = self.detect(img, landmarks=True)
# Select faces
if not self.keep_all:
    img, batch_boxes, batch_probs, batch_points = self.select_boxes(
        batch_boxes, batch_probs, batch_points, img, method=self.selection_method,
    )
# Extract faces
faces = self.extract(img, batch_boxes, save_path)

if return_prob:
    return faces, batch_probs
else:
    return faces

if batch_mode:
    sanitized_selected_boxes = []
    sanitized_selected_probs = []
    sanitized_selected_points = []
    for idx, box in enumerate(selected_boxes):
        if box is None:
            if isinstance(imgs, list):
                imgs.pop(idx)
            elif isinstance(imgs, np.ndarray):
                imgs = np.delete(imgs, idx, 0)
            elif isinstance(imgs, torch.Tensor):
                imgs = torch.cat((imgs[:idx], imgs[idx+1:]), 0)
            else:
                raise TypeError("imgs must be a list, np.ndarray, or torch.Tensor")
        else:
            sanitized_selected_boxes.append(box)
            sanitized_selected_probs.append(selected_probs[idx])
            sanitized_selected_points.append(selected_points[idx])
    selected_boxes = np.array(sanitized_selected_boxes)
    selected_probs = np.array(sanitized_selected_probs)
    selected_points = np.array(sanitized_selected_points)
else:
    selected_boxes = selected_boxes[0]
    selected_probs = selected_probs[0][0]
    selected_points = selected_points[0]

return imgs, selected_boxes, selected_probs, selected_points