【无标题】

import matplotlib.pyplot as plt
import tensorflow.keras
from mtcnn.mtcnn import MTCNN
from matplotlib.patches import Rectangle
import cv2 as cv2
import pandas as pd

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession

config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

img = plt.imread('xz.jpg')

model = MTCNN()

faces = model.detect_faces(img)

for face in faces:
    print(face)

def draw_boxes(img_path,faces):
    img = plt.imread(img_path)

    plt.imshow(img)

    plt.show()

    ax = plt.gca()

    for face in faces:
        x,y,w,h = face['box']
        rect = Rectangle((x,y),w,y,fill=False,color='blue')
        ax.add_patch(rect)

        for k,v in face['Keypoints'].item():
            point = Circle(v,redius=2,color='red')
            ax.add_patch(point)

    plt.show()


draw_boxes(img, faces)