在做人脸检测时,我们需要将检测的结果绘制到图片上查看效果,这时可以利用 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<FaceInfo> 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<float>(i, j);//特征点在图像中横坐标
point.y = face_key_pionts.at<float>(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