keras 在tensorflow 2.0中是否有mtcnn人脸检测的实现？

cbwuti44 于 2023-05-18 发布在其他

关注(0)|答案(2)|浏览(108)

最近我已经转移到tensorflow==2.0.0-rc 0，现在用于人脸检测的mtcnn在我的电脑上不工作。我可以找到tensorflow==2.0.0-rc 0版本的mtcnn吗？纯Keras实现的mtcnn也可以在这种情况下工作。
我尝试过在https://github.com/nyoki-mtl/keras-facenet上实现facenet的keras。它在kears中很好地实现了facenet，但缺少人脸检测部分（keras中的mtcnn）。

keras

来源：https://stackoverflow.com/questions/57757515/is-there-any-implementation-of-mtcnn-face-detection-in-tensorflow-2-0

2条答案

按热度按时间

jogvjijk1#

以下是真实的运行MTCNN face_detection的代码：

import torch
from torchvision import transforms 
from PIL import Image 
import numpy as np 
import cv2 
from mtcnn import MTCNN

cap = cv2.VideoCapture(0)
detector = MTCNN()

while True:

    ret,frame = cap.read()

    output = detector.detect_faces(frame)

    for single_output in output:
        x,y,w,h = single_output['box']
        cv2.rectangle(frame,pt1=(x,y), pt2=(x+w,y+h),color=(255,0,0),thickness=2)
    cv2.imshow('win',frame)

    if cv2.waitKey(1) & 0xFF == ('x'):
        break

cv2.destroyAllWindows()

赞(0）回复(0）举报 2023-05-18

6za6bjd02#

我也需要TF 2，所以只是推了this library。您应该能够克隆repo和python setup.py install来安装它。它是针对TensorFlow 2.5.0开发的。
P.S.对于那些希望在未来将TF 1.x库移植到TF 2的人来说，这条评论可能是你所需要的！

赞(0）回复(0）举报 2023-05-18