OpenCV 绘制人脸检测框和关键点



在做人脸检测时,我们需要将检测的结果绘制到图片上查看效果,这时可以利用 OpenCV 的基本绘图函数将人脸检测框和人脸关键点绘制到原图上。

关键代码如下:

    // TODO FaceDetectEngine 初始化
    cv::Mat image = cv::imread(test_file, cv::IMREAD_UNCHANGED);// 原始图像
    cv::Mat gray;
    cv::cvtColor(image, gray, cv::COLOR_BGR2GRAY);// 灰度图

    // 检测出人脸框
    std::vector face_info_list = FaceDetectEngine.detect(gray.data, gray.cols, gray.rows);
    int face_num = face_info_list.size();
    for (int i = 0; i < face_num; ++i)
    {
        FaceInfo faceInfo = face_info_list[i];
        float sketch_face_info[5] = {
                faceInfo.x * 1.0f / gray.cols,
                faceInfo.y * 1.0f / gray.rows,
                faceInfo.width * 1.0f / gray.cols,
                faceInfo.height * 1.0f / gray.rows,
                faceInfo.rotation
        };
        // 检测人脸关键点
        cv::Mat face_key_pionts = FaceDetectEngine.detectKeyPoints(gray, sketch_face_info);

        cv::Point leftTopP;
        leftTopP.x = faceInfo.x;
        leftTopP.y = faceInfo.y;

        cv::Point rightBottomP;
        rightBottomP.x = faceInfo.x + faceInfo.width;
        rightBottomP.y = faceInfo.y + faceInfo.height;

        // 绘制人脸框,cvRectangle函数参数: 图片, 左上角, 右下角, 颜色, 线条粗细, 线条类型,点类型
        cv::rectangle(image, leftTopP, rightBottomP, cv::Scalar(0, 0, 255), 1, 4, 0);

        // 绘制 68 点的人脸关键点
        for (int i = 0; i < face_key_pionts.rows; i+=2) {
            for (int j = 0; j < face_key_pionts.cols; ++j) {

                cv::Point point;//特征点,用以画在图像中
                point.x = face_key_pionts.at(i, j);//特征点在图像中横坐标
                point.y = face_key_pionts.at(i + 1, j);//特征点在图像中纵坐标
                cv::circle(image, point, 1, cv::Scalar(0, 255, 0), 2);//在图像中画出特征点,1是圆的半径
            }
        }
    }

    // 保存图片
    FILE * file = fopen(save_path, "wb");
    fwrite(image.data, image.elemSize(), image.cols * image.rows, file);
    fclose(file);

    // TODO 释放 FaceDetectEngine 

你可能感兴趣的:(原创,C/C++,OpenCV)