from facenet_pytorch import MTCNN
import cv2
from PIL import Image, ImageDraw, ImageColor
import numpy as np
from matplotlib import pyplot as plt

mtcnn = MTCNN(keep_all=True)

img = Image.open("face.jpg")

boxes, _ = mtcnn.detect(img)
draw = ImageDraw.Draw(img)
for i, box in enumerate(boxes):
    draw.rectangle(((box[0], box[1]), (box[2], box[3])), outline="red")
img.save("detected.jpg")
#print(face)