欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

关于matlab 中libsvm中model中的保存与调用新发现 博客分类: matlab 图像 matlablibsvmsavemodelloadmodel

程序员文章站 2024-02-08 14:53:10
...

代码在网上可以找到,savemodel.c与loadmodel.c 我把网上的引用放在下面

 

最近一直在用matlab和libsvm,发现libsvm库用起来还是很方便的,就是没有模型直接保存到文件和读取模型的matlab接口(C++的接口有)。由于有会用的Opencv等C/C++库,所以数据交换比较麻烦。看了一下libsvm的svm.h、svm.cpp文件,发现有svm_save_model(),svm_load_model()等函数。于是乎用mex小做封装,写了两个matlab可以直接调用的接口。

 

保存svm model到文件:(savemodel.c)

 

[cpp] view plaincopy
 
  1. #include <stdio.h>  
  2. #include <stdlib.h>  
  3. #include <string.h>  
  4. #include <ctype.h>  
  5. #include "svm.h"  
  6.   
  7. #include "mex.h"  
  8. #include "svm_model_matlab.h"  
  9.   
  10.   
  11. void exit_with_help()  
  12. {  
  13.     mexPrintf(  
  14.     "Usage: savemodel('filename', model);\n"  
  15.     );  
  16. }  
  17.   
  18. int savemodel(const char *filename, const mxArray *matlab_struct)  
  19. {  
  20.     const char *error_msg;  
  21.     struct svm_model* model;  
  22.     int result;  
  23.     model = matlab_matrix_to_model(matlab_struct, &error_msg);  
  24.       
  25.     if (model == NULL)  
  26.     {  
  27.         mexPrintf("Error: can't read model: %s\n", error_msg);  
  28.     }  
  29.       
  30.     result = svm_save_model(filename, model);  
  31.       
  32.     if( result != 0 )  
  33.     {  
  34.         mexPrintf("Error: can't write model to file!\n");  
  35.     }  
  36.       
  37.     return result;  
  38. }  
  39.   
  40. void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])  
  41. {  
  42.     if(nrhs == 2)  
  43.     {  
  44.         char filename[256];  
  45.         int *result;  
  46.           
  47.         mxGetString(prhs[0], filename, mxGetN(prhs[0])+1);  
  48.           
  49.         plhs[0] = mxCreateNumericMatrix(1, 1, mxINT8_CLASS, 0);  
  50.           
  51.         result = mxGetPr(plhs[0]);  
  52.         *result = savemodel(filename, prhs[1]);  
  53.     }  
  54.     else  
  55.     {  
  56.         exit_with_help();         
  57.         return;  
  58.     }  
  59. }  


读取文件中的svm model:( loadmodel.c )

 

 

[cpp] view plaincopy
 
  1. #include <stdio.h>  
  2. #include <stdlib.h>  
  3. #include <string.h>  
  4. #include <ctype.h>  
  5. #include "svm.h"  
  6.   
  7. #include "mex.h"  
  8. #include "svm_model_matlab.h"  
  9.   
  10.   
  11. void exit_with_help()  
  12. {  
  13.     mexPrintf(  
  14.     "Usage: model = loadmodel('filename', num_of_feature);\n"  
  15.     );  
  16. }  
  17.   
  18. void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])  
  19. {  
  20.     if(nrhs == 2)  
  21.     {  
  22.         char filename[256];  
  23.         int num_of_feature;  
  24.         struct svm_model* model;  
  25.         int featurenum;  
  26.         const char *error_msg;  
  27.           
  28.         mxGetString(prhs[0], filename, mxGetN(prhs[0])+1);  
  29.         model = svm_load_model(filename);  
  30.           
  31.         if(model == NULL)  
  32.         {  
  33.             mexPrintf("Error: can't read the model file!\n");  
  34.             return;  
  35.         }  
  36.           
  37.         featurenum = *(mxGetPr(prhs[1]));  
  38.           
  39.         error_msg = model_to_matlab_structure(plhs, featurenum, model);  
  40.           
  41.         if(error_msg)  
  42.             mexPrintf("Error: can't convert libsvm model to matrix structure: %s\n", error_msg);  
  43.           
  44.         svm_free_and_destroy_model(&model);  
  45.     }  
  46.     else  
  47.     {  
  48.         exit_with_help();         
  49.         return;  
  50.     }  
  51. }  


这两个文件放入libsvm-3.1/matlab目录下,然后打开同目录下的make.m文件,添加如下两行(红色部分):

 

mex -O -largeArrayDims -I..\ -c ..\svm.cpp
mex -O -largeArrayDims -I..\ -c svm_model_matlab.c
mex -O -largeArrayDims -I..\ svmtrain.c svm.obj svm_model_matlab.obj
mex -O -largeArrayDims -I..\ svmpredict.c svm.obj svm_model_matlab.obj
mex -O -largeArrayDims -I..\ savemodel.c svm.obj svm_model_matlab.obj
mex -O -largeArrayDims -I..\ loadmodel.c svm.obj svm_model_matlab.obj

最后在matlab中,libsvm-3.1/matlab目录下运行“make”,等到编译完成后,生成savemodel.mexw32、loadmodel.mexw32两个文件。(64位的系统生成的两个文件扩展名不一样)

此时就可以在matlab中使用savemodel('filename', model)和model = loadmodel('filename', num)来读取和保存训练好的svm模型了。

注意:loadmodel接口中的第二个参数num是向量的维度(特征数),即500个30维的样本,num=30

(貌似多了这个参数是因为matlab调用了model_to_matlab_structure这个接口的缘故,在c/c++中读写模型就不需用num这个参数了)

 

如果需要在c/c++程序中调用libsvm的模型文件,就直接用svm_save_model(),svm_load_model()等函数进行操作了。

 

我用的版本是libsvm 2.91,savemodel.c没有问题,loadmodel需要做如下修改

svm_free_and_destroy_model(&model);  修改为   svm_destroy_model(model);

 

完美运行