OpenCV人脸检测使用Caffe的训练模型

王小鹏鹏

已于 2025-02-20 17:32:11 修改

阅读量2.1k

点赞数

分类专栏： OpenCV 文章标签： opencv caffe 人工智能

于 2019-01-16 16:55:47 首次发布

本文链接：https://blog.youkuaiyun.com/kingroc/article/details/86510791

版权

OpenCV 专栏收录该内容

78 篇文章

订阅专栏

本文是OpenCV系列教程的一部分，介绍如何利用Caffe训练的人脸检测模型，在OpenCV中进行人脸检测。提供了具体的代码实现位于105-FindFacebyCaffeModel.py。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

《OpenCV系列教程》
项目位置：OpenCV-Sample
代码位置：105-FindFacebyCaffeModel.py
在这里插入图片描述
代码：

import cv2 as cv
from cv2 import dnn

cap = cv.VideoCapture('./res/test.mp4')
net = dnn.readNetFromCaffe('./res/deploy.prototxt', './res/res10_300x300_ssd_iter_140000.caffemodel')

inWidth = 300
inHeight = 300
confThreshold = 0.5

while True:
    ret, frame = cap.read()
    cols = frame.shape[1]
    rows = frame.shape[0]

    net.setInput(dnn.blobFromImage(frame, 1.0, (inWidth, inHeight), (104.0, 177.0, 123.0), False, False))
    detections = net.forward()
    # perf_stats = net.getPerfProfile()
    # print('Inference time, ms: %.2f' % (perf_stats[0] / cv.getTickFrequency() * 1000))
    for i in range(detections.shape[2]):
        confidence = detections[0, 0, i, 2]
        if confidence > confThreshold:
            xLeftBottom = int(detections[0, 0, i, 3] * cols)
            yLeftBottom = int(detections[0, 0, i, 4] * rows)
            xRightTop = int(detections[0, 0, i, 5] * cols)
            yRightTop = int(detections[0, 0, i, 6] * rows)

            cv.rectangle(frame, (xLeftBottom, yLeftBottom), (xRightTop, yRightTop), (0, 0, 255))
            if cv.waitKey(30) == 27:
                break;

    cv.imshow("detections", frame)
    if cv.waitKey(1) == 27:
        break

cv.destroyAllWindows()