我正在使用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++;
}

07-27 19:51