我正在使用OpenCV检测多个图像中的面部,然后将面部图像(矩形)保存为jpg文件。但是由于有很多镜像,所以我收到内存不足的OpenCV错误。我知道这是因为使用了不会释放自身的IplImage
,我应该改用cv::Mat
,但是我找不到使用cv::Mat
想要解决的解决方案。有人可以帮忙吗?
这是我的代码
int main (int argc, const char* argv[] )
{
cv::CascadeClassifier face_cascade;
face_cascade.load("haarcascade_frontalface_alt.xml");
cv::Mat captureFrame;
cv::Mat grayscaleFrame;
DIR *dir;
struct dirent *ent;
if ((dir = opendir ("c:\\images\\")) != NULL) {
int counter = 1;
while ((ent = readdir (dir)) != NULL) {
printf ("%s\n", ent->d_name);
std::string fullPath = std::string("c:\\images\\") + ent->d_name;
captureFrame = cv::imread(fullPath);
cvtColor(captureFrame, grayscaleFrame, CV_BGR2GRAY);
equalizeHist(grayscaleFrame, grayscaleFrame);
std::vector<cv::Rect> faces;
face_cascade.detectMultiScale(grayscaleFrame, faces, 1.2,2,CV_HAAR_DO_CANNY_PRUNING, cv::Size(60, 60));
for(int i = 0; i < faces.size(); i++)
{
IplImage* img = cvCloneImage(&(IplImage)captureFrame);
cvSetImageROI(img, cvRect(faces.at(i).x, faces.at(i).y, faces.at(i).width, faces.at(i).height));
IplImage *img2 = cvCreateImage(cvGetSize(img),
img->depth,
img->nChannels);
cvCopy(img, img2, NULL);
cvResetImageROI(img);
std::stringstream sstm;
sstm << "faces\\" <<counter << ".jpg";
string result = sstm.str();
cvSaveImage (result.c_str() , img2);
counter++;
}
}
closedir (dir);
}
return 1;
}
最佳答案
摆脱这种局面的好主意!
看,它也变得容易得多:
for(size_t i = 0; i < faces.size(); i++)
{
Mat roi(captureFrame, faces[i]);
std::stringstream sstm;
sstm << "faces\\" <<counter << ".jpg";
string result = sstm.str();
imwrite(result.c_str() , roi);
counter++;
}