OpenCv SVM 输出文件格式 [英] OpenCv SVM output file format

问题描述

我正在实现自己的 SVM，而不是使用 OpenCV 的 svm 类.我想要我的 SVM 用来保存其输出的 XML 文件，如果我愿意的话，将来可以由 OpenCV 的 SVM 加载和使用.我需要为此做什么?

I am implementing my own SVM rather than using OpenCV's svm class. I want the XML file that my SVM uses to save its output, can be loaded and used by OpenCV's SVM in future, if I wish. What should I need to do for that ?

简而言之:OpenCV 用于存储其 SVM 输出的格式是什么?

In short : what is the format that OpenCV uses to store its SVM output ?

推荐答案

可以参考 OpenCV 的 CvSVM::write 函数.

You can follow OpenCV's CvSVM::write function.

int i, var_count = get_var_count(), df_count, class_count;
const CvSVMDecisionFunc* df = decision_func;

cvStartWriteStruct( fs, name, CV_NODE_MAP, CV_TYPE_NAME_ML_SVM );

write_params( fs );

cvWriteInt( fs, "var_all", var_all );
cvWriteInt( fs, "var_count", var_count );

class_count = class_labels ? class_labels->cols :
              params.svm_type == CvSVM::ONE_CLASS ? 1 : 0;

if( class_count )
{
    cvWriteInt( fs, "class_count", class_count );

    if( class_labels )
        cvWrite( fs, "class_labels", class_labels );

    if( class_weights )
        cvWrite( fs, "class_weights", class_weights );
}

if( var_idx )
    cvWrite( fs, "var_idx", var_idx );

// write the joint collection of support vectors
cvWriteInt( fs, "sv_total", sv_total );
cvStartWriteStruct( fs, "support_vectors", CV_NODE_SEQ );
for( i = 0; i < sv_total; i++ )
{
    cvStartWriteStruct( fs, 0, CV_NODE_SEQ + CV_NODE_FLOW );
    cvWriteRawData( fs, sv[i], var_count, "f" );
    cvEndWriteStruct( fs );
}

cvEndWriteStruct( fs );

// write decision functions
df_count = class_count > 1 ? class_count*(class_count-1)/2 : 1;
df = decision_func;

cvStartWriteStruct( fs, "decision_functions", CV_NODE_SEQ );
for( i = 0; i < df_count; i++ )
{
    int sv_count = df[i].sv_count;
    cvStartWriteStruct( fs, 0, CV_NODE_MAP );
    cvWriteInt( fs, "sv_count", sv_count );
    cvWriteReal( fs, "rho", df[i].rho );
    cvStartWriteStruct( fs, "alpha", CV_NODE_SEQ+CV_NODE_FLOW );
    cvWriteRawData( fs, df[i].alpha, df[i].sv_count, "d" );
    cvEndWriteStruct( fs );
    if( class_count > 1 )
    {
        cvStartWriteStruct( fs, "index", CV_NODE_SEQ+CV_NODE_FLOW );
        cvWriteRawData( fs, df[i].sv_index, df[i].sv_count, "i" );
        cvEndWriteStruct( fs );
    }
    else
        CV_ASSERT( sv_count == sv_total );
    cvEndWriteStruct( fs );
}
cvEndWriteStruct( fs );
cvEndWriteStruct( fs );

以下是创建 CvFileStorage 并将其保存到磁盘的方法:

And here is how you can create a CvFileStorage and save it to the disk:

const char* filename = "/xxx/yyy/zzz";
const char* modelname = "svm";
CvFileStorage* fs = cvOpenFileStorage(filename, 0, CV_STORAGE_WRITE);
if (fs) {
  write(fs, modelname);
}
cvReleaseFileStorage(&fs);

通过这种方式，您可以选择将模型保存为 XML 或 YAML 格式，并使用 CvSVM::read() 加载模型文件.希望这会有所帮助.

By doing this way, you can choose to save the model into XML or YAML format, and use CvSVM::read() to load the model file. Hope this helps.

这篇关于OpenCv SVM 输出文件格式的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！