MTCNN切割人脸

本文介绍了如何利用facenet_pytorch库实现MTCNN进行人脸检测,包括环境配置、代码示例和实际操作,展示了在CPU和GPU环境下的人脸框绘制结果。
摘要由CSDN通过智能技术生成

使用facenet_pytorch已经封装好了的库

环境:

pip install facenet_pytorch

代码:

from facenet_pytorch import MTCNN
import cv2

mtcnn = MTCNN()
img = cv2.imread('./pic_3.jpg')

#输出三元素,矩形框、概率、关键点
boxes, probs ,points= mtcnn.detect(img,landmarks=True)

#输出二元素,矩形框、概率
boxes2, probs2 = mtcnn.detect(img)

for index, bbox in enumerate(boxes):
    if bbox is not None:
        ymin, xmin, ymax ,xmax
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值