效果图
先来看看效果图吧。
检测后的结果将会显示在文件夹里。
环境配置
- 首先,需要配置opencv环境,可以参考这篇文章:Windows下OpenCV 3.4.0 + Visual Studio
2015开发环境的配置 - 其次,是核心识别库环境配置,这里放上另一篇文章,配置步骤写的很详细可以参考一下:opencv核心识别库环境配置
代码
一 、简述
使用MFC绘制对话框就不过多赘述了,这里主要讲一下“打开目标图片”,“裁剪图片”,“还原图片”,“检测图片”这四个功能是如何实现的。
二、打开目标图片
void CFaceRecognitionDlg::OnBnClickedButtonInputTarget()//打开图片控件
{
// TODO: 在此添加控件通知处理程序代码
//打开文件 定义四种格式的文件bmp gif jpg tiff
CString filter;
filter = "所有文件(*.bmp,*.jpg,*.gif,*tiff)|*.bmp;*.jpg;*.gif;*.tiff| BMP(*.bmp)|*.bmp| JPG(*.jpg)|*.jpg| GIF(*.gif)|*.gif| TIFF(*.tiff)|*.tiff||";
CFileDialog dlg(TRUE, NULL, NULL, OFN_HIDEREADONLY, filter, NULL);
//按下确定按钮 dlg.DoModal() 函数显示对话框
if (dlg.DoModal() == IDOK)
{
//打开对话框获取图像信息
BmpName = dlg.GetPathName(); //获取文件路径名 如D:\pic\abc.bmp
CString EntName = dlg.GetFileExt(); //获取文件扩展名
EntName.MakeLower(); //将文件扩展名转换为一个小写字符
IplImage* srcimage;
srcimage = cvLoadImage(BmpName);//图片路径
ShowImage(srcimage, IDC_STATIC_PIC);//显示图片
DeleteDirectory("..\\picture\\");//删除该目录下的文件
Mat ImgTemp = cvarrToMat(srcimage);//IplImage转为Mat
Mat Img = ImgTemp.clone();
cv::imwrite("..\\picture\\dst.bmp", Img);//将图片保存
}
}
- 每次打开图片会将图片显示到对话框中。
- 删除"…\picture\"路径中的图片,此路径用于存储打开后的图片、裁剪后的图片、检测结果。
- 将打开后的图片存储为"…\picture\dst.bmp",方便后续对图片的处理
上述代码还使用到了显示图片函数和删除图片函数
1.显示图片函数
void CFaceRecognitionDlg::ShowImage(IplImage* img, UINT ID)//显示图片
{
CDC* pDC = GetDlgItem(ID)->GetDC(); //获得显示控件的DC
HDC hDC = pDC->GetSafeHdc(); //获得HDC(设备句柄)来进行绘图操作
CRect rect;
GetDlgItem(ID)->GetClientRect(&rect);
SetRect(rect, rect.left, rect.top, rect.right, rect.bottom);
CvvImage cimg;
cimg.CopyOf(img); //复制图片
cimg.DrawToHDC(hDC, &rect); //将图片绘制到显示控件的指定区域内
ReleaseDC(pDC);
}
2.删除图片函数
BOOL DeleteDirectory(const CString &strPath)//删除文件夹中所有文件
{
CFileFind tempFind;
TCHAR sTempFileFind[MAX_PATH] = { 0 };
wsprintf(sTempFileFind, _T("%s\\*.*"), strPath);
BOOL IsFinded = tempFind.FindFile(sTempFileFind);
while (IsFinded)
{
IsFinded = tempFind.FindNextFile();
if (!tempFind.IsDots())
{
TCHAR sFoundFileName[200] = { 0 };
_tcscpy(sFoundFileName, tempFind.GetFileName().GetBuffer(200));
if (tempFind.IsDirectory())
{
TCHAR sTempDir[200] = { 0 };
wsprintf(sTempDir, _T("%s\\%s"), strPath, sFoundFileName);
DeleteDirectory(sTempDir); //删除文件夹下的文件
RemoveDirectory(sTempDir); //移除空文件
}
else
{
TCHAR sTempFileName[200] = { 0 };
wsprintf(sTempFileName, _T("%s\\%s"), strPath, sFoundFileName);
DeleteFile(sTempFileName);
}
}
}
tempFind.Close();
// if(!RemoveDirectory(strPath))
// return false;
return true;
}
三、裁剪图片
图片裁剪后将会在对话框中重新显示图片,并保存为dst.bmp,下一步检测图片中的人脸时将会只检测剪裁后的图片中的人脸。
void CFaceRecognitionDlg::OnBnClickedButtonShear() //裁剪图片按钮
{
// TODO: 在此添加控件通知处理程序代码
IplImage* srcimage;
srcimage = cvLoadImage(BmpName);//图片路径
IplImage* img_show = cvCloneImage(srcimage); //复制打开后的图片为后续处理准备
Mat ImgTemp;
Mat Img;
cvNamedWindow("裁剪图片",0);
cvSetMouseCallback("裁剪图片", cvMouseCallback);
while (true)
{
cvCopy(srcimage, img_show);
cvRectangle(img_show, pt1, pt2, cvScalar(0, 0, 255), 4);
cvShowImage("裁剪图片", img_show);
char key = cvWaitKey(5);
if (key == '\r')
{
cvSetImageROI(img_show, cvRect(pt1.x, pt1.y, abs(pt2.x - pt1.x), abs(pt2.y - pt1.y)));
DeleteDirectory("..\\picture\\");//删除该目录下的文件
ImgTemp = cvarrToMat(img_show);//IplImage转为Mat
Img = ImgTemp.clone();
cv::imwrite("..\\picture\\dst.bmp", Img);//将裁剪好的图片保存
ShowImage(img_show, IDC_STATIC_PIC);
break;
}
if (!cvGetWindowHandle("裁剪图片"))//关闭窗口
{
break;
}
}
cvReleaseImage(&srcimage);
cvReleaseImage(&img_show);
cvDestroyWindow("裁剪图片");
}
四、还原图片
函数中使用到的BmpName变量是CString全局变量,在打开图片函数中已经赋值,打开图片时是原始图片还未进行裁剪,所以还原图片时,只需要再次使用该变量,将裁剪后的图片覆盖掉,就能实现还原图片的功能了。
void CFaceRecognitionDlg::OnBnClickedButtonRestorePic()//重置图片
{
// TODO: 在此添加控件通知处理程序代码
IplImage* srcimage;
Mat ImgTemp;
Mat Img;
srcimage = cvLoadImage(BmpName);//图片路径
ShowImage(srcimage, IDC_STATIC_PIC);
DeleteDirectory("..\\picture\\");//删除该目录下的文件
ImgTemp = cvarrToMat(srcimage);//IplImage转为Mat
Img = ImgTemp.clone();
cv::imwrite("..\\picture\\dst.bmp", Img);//将裁剪好的图片保存
ShowImage(srcimage, IDC_STATIC_PIC);
}
五、检测图片
检测出的人脸图片将会按序号保存到…\picture文件夹中。
void CFaceRecognitionDlg::OnBnClickedButtonFindFaces()//图片人脸检索
{
// TODO: 在此添加控件通知处理程序代码
Mat src = imread("..\\picture\\dst.bmp");
Mat frame = src.clone();
Mat facesRIO;
//图像缩放,采用双线性插值。
//cv::resize(src,src,Size(128,128),0,0,cv::INTER_LINEAR);
//图像灰度化。
cv::cvtColor(src, src, COLOR_BGR2GRAY);
//直方图均衡化,图像增强,暗的变亮,亮的变暗。
cv::equalizeHist(src, src);
//
String face_cascade_name = "..\\xml\\haarcascade_frontalface_alt.xml";
String eyes_cascade_name = "..\\xml\\haarcascade_eye_tree_eyeglasses.xml";
CascadeClassifier face_cascade, eyes_cascade;
if (!face_cascade.load(face_cascade_name))
{
//加载脸部分类器失败!
return;
}
if (!eyes_cascade.load(eyes_cascade_name))
{
//加载眼睛分类器失败!
return;
}
CString targetSrcStr;
//存储找到的脸部矩形。
std::vector<Rect> faces;
face_cascade.detectMultiScale(src, faces, 1.1, 2, 0 | CASCADE_SCALE_IMAGE, Size(30, 30));
for (size_t i = 0; i < faces.size(); ++i)
{
//绘制矩形 BGR。
rectangle(frame, faces[i], Scalar(0, 0, 255), 1);
//截取人脸。
facesRIO = frame(faces[i]);
//图像缩放。
cv::resize(facesRIO, facesRIO, Size(128, 128), 0, 0, cv::INTER_LINEAR);
//保存图像。
USES_CONVERSION;
targetSrcStr.Format(_T("..\\picture\\%lld.jpg"), i);
cv::String cvSrcStr =targetSrcStr;
cv::imwrite(cvSrcStr, facesRIO);// 保存图片
}
return;
}