基于SVM+HOG的花生品种识别

早在大二时，导师做过一个花生品种识别的程序，但当时用于品种识别的特征是自定义特征；而我的当初的本科毕业设计则是基于SVM + HOG的人脸识别；在后来的工作学习中，又用到了HOG特征，想着是不是之前的花生品种的识别也能用Hog特征进行识别；正所谓学以致用，也能巩固下刚刚所学的知识；

所谓的Hog特征与SVM网上资料一大堆，而且讲解得非常详细，在这里不在做过多的讲解；

本程序的开发环境，主要是依赖于VS2013 + OpenCV2.4.9，开发环境，请自行配置，那么，下面贴出代码，不足之处，请各位大侠不吝批评指正：

特征提取功能的实现：

CvMat* dataMat = NULL;
Book* book = xlCreateBook();
Sheet* sheet = book->addSheet("Sheet1");

for (int j = 1; j <= 13; ++ j)
{
    CString strLoadPath = "SubImage\";
    CString strFile;
    strFile.Format("s%d", j);
    strLoadPath = strLoadPath + strFile + "\";

    for (int i = 1; i <= 45; ++i)
    {
        CString loadPath = strLoadPath;
        CString stri;
        stri.Format("%d.jpg", i);
        loadPath = loadPath + stri;

        IplImage* srcImage = cvLoadImage(loadPath);
        IplImage* grayImage = cvCreateImage(cvGetSize(srcImage), IPL_DEPTH_8U, 1);
        IplImage* hogImage = cvCreateImage(cvSize(96, 64), IPL_DEPTH_8U, 1);        //用于提取Hog特征的图像

        cvCvtColor(srcImage, grayImage, CV_RGB2GRAY);

        cvResize(grayImage, hogImage, CV_INTER_LINEAR);   //线性插值

        std::vector<float> vfDescriptors;
        //vfDescriptors.resize(hogImage->width * hogImage->height);
        cv::Ptr<cv::HOGDescriptor> hog = new cv::HOGDescriptor(cvSize(hogImage->width, hogImage->height), cvSize(16, 16), cvSize(8, 8), cvSize(8, 8), 9);
        hog->compute(hogImage, vfDescriptors, cv::Size(1, 1), cv::Size(0, 0));       //使用计算函数进行计算

        //CvMat* mat = cvCreateMat(45 * 13, vfDescriptors.size(), CV_32FC1);
        //cvZero(mat);

        int cols = 0;

        for (auto it = vfDescriptors.begin(); it != vfDescriptors.end(); ++it)
        {
            if (sheet)
            {
                sheet->writeNum(45 * (j-1) + (i - 1), cols++, *it);    //把Hog数据存储到Excel 中
            }
        }

        //释放内存
        cvReleaseImage(&srcImage);
        srcImage = NULL;
        cvReleaseImage(&grayImage);
        grayImage = NULL;
        cvReleaseImage(&hogImage);
        hogImage = NULL;

    }// 内层 for 循环

}

book->save("Hog.xls");
book->release();

SVM训练的实现代码：

Book* book = xlCreateBook();

    if (book->load("Hog.xls"))
    {
        Sheet* sheet = book->getSheet(0);

        if (sheet)
        {

            CvMat* dataMat = cvCreateMat(sheet->lastRow(), sheet->lastCol(), CV_32FC1);       //存储导入的数据

            for (int i = 0; i < sheet->lastRow(); ++ i)
            {
                for (int j = 0; j < sheet->lastCol(); ++ j)
                {
                    double temp = sheet->readNum(i, j);
                    cvSetReal2D(dataMat, i, j, temp);
                }
            }

            MessageBox("数据导入完成");

            CvMat* lableMat = cvCreateMat(sheet->lastRow(), 1, CV_32FC1);
            cvZero(lableMat);

            for (int i = 0; i < 13; ++ i)
            {
                for (int j = 0; j < 45; ++ j)
                {
                    cvSetReal2D(lableMat, i * 45 + j, 0, i + 1);
                }
            }

            CvSVM svm;
            CvSVMParams svmParams;                      //向量机参数
            CvTermCriteria  svmCriteria;                //迭代中止条件
            svmCriteria = cvTermCriteria(CV_TERMCRIT_EPS, 1000, FLT_EPSILON);

            svmParams = CvSVMParams(CvSVM::C_SVC, CvSVM::RBF, 10.0, 8.0, 1.0, 10.0, 0.5, 0.1, NULL, svmCriteria);

            //利用训练数据和确定的学习参数，进行SVM学习
            svm.train(dataMat, lableMat, NULL, NULL, svmParams);
            svm.save("svm.xml");

            //以下代码用于SVM识别
            CvMat* svmResult = cvCreateMat(sheet->lastRow(), 1, CV_32FC1);
            CvMat* svmRow = NULL;

            for (int i = 0; i < sheet->lastRow(); ++ i)
            {
                svmRow = cvCreateMat(1, sheet->lastCol(), CV_32FC1);

                for (int j = 0; j < sheet->lastCol(); ++ j)
                {
                    float temp = cvGetReal2D(dataMat, i, j);
                    cvSetReal2D(svmRow, 0, j, temp);
                }

                unsigned int ret = 0;
                ret = svm.predict(svmRow);
                cvSetReal2D(svmResult, i, 0, ret);

                cvReleaseMat(&svmRow);
                svmRow = NULL;
            }

            int nCount = 0;

            for (int i = 0; i < 13; ++i)
            {
                for (int j = 0; j < 45; ++ j)
                {
                    int ret = cvGetReal2D(svmResult, i * 45 + j, 0);
                    if (ret == (i + 1))
                    {
                        ++nCount;
                    }
                }
            }

            float recognize = 100 * nCount / 13 / 45;

            CString str;
            str.Format("SVM 识别率为： %f", recognize);
            str = str + "%";
            MessageBox(str);

        }
    }

测试功能的实现代码：

CvSVM svm;
    svm.load("svm.xml");

    CFileDialog dlg(TRUE, NULL, NULL, 0, "图片文件(*.jpg)|*.jpg||");
    if (dlg.DoModal() == IDOK)
    {
        IplImage* testImage = cvLoadImage(dlg.GetPathName());
        IplImage* grayImage = cvCreateImage(cvGetSize(testImage), IPL_DEPTH_8U, 1);
        IplImage* hogImage = cvCreateImage(cvSize(96, 64), IPL_DEPTH_8U, 1);

        cvCvtColor(testImage, grayImage, CV_RGB2GRAY);
        cvResize(grayImage, hogImage, CV_INTER_LINEAR);

        std::vector<float> vfDescriptors;
        //vfDescriptors.resize(hogImage->width * hogImage->height);
        cv::Ptr<cv::HOGDescriptor> hog = new cv::HOGDescriptor(cvSize(hogImage->width, hogImage->height), cvSize(16, 16), cvSize(8, 8), cvSize(8, 8), 9);
        hog->compute(hogImage, vfDescriptors, cv::Size(1, 1), cv::Size(0, 0));       //使用计算函数进行计算

        CvMat* mat = cvCreateMat(1, 256, CV_32FC1);

        int cols = 0;

        for (auto it = 0; it < 256; ++it)
        {
            cvSetReal2D(mat, 0, it, vfDescriptors[it]);
        }


        int  ret = svm.predict(mat);
        CString str;
        switch (ret)
        {
        case 1:
            str = "品种1：P12";
            break;
        case 2:
            str = "品种2：矮2";
            break;
        case 3:
            str = "品种3：花玉22";
            break;
        case 4:
            str = "品种4：花玉25";
            break;
        case 5:
            str = "品种5：冀花2号";
            break;
        case 6:
            str = "品种6：冀花4号";
            break;
        case 7:
            str = "品种7：冀花5号";
            break;
        case 8:
            str = "品种8：鲁花9号";
            break;
        case 9:
            str = "品种9：青花6号";
            break;
        case 10:
            str = "品种10：天府3号";
            break;
        case 11:
            str = "品种11：维花8号";
            break;
        case 12:
            str = "品种12：小白沙";
            break;
        case 13:
            str = "品种13：中农108";
            break;
        }
        AfxMessageBox(str);
        //cvReleaseMat(&mat);
    }//end if

相关的程序代码，用于测试的图像样本及程序中用到的操作 Excel 的类，已经打包上传到CSDN，请自行下载：http://download.csdn.net/detail/lingtianyulong/8377461