在把 Matlab 轉(zhuǎn)化成 OpenCV（C++） 的過程中癣猾，經(jīng)常會遇見 load .mat 文件這個問題青灼。正巧暂吉，Stackoverflow 上有個問題《Converting a .mat file from MATLAB into cv::Mat matrix in OpenCV》就是回答這個的竣付。這篇博文的主要分為兩部分，第 1 部分是將這個答案轉(zhuǎn)述一遍竭钝，以方便日后需要；第 2 部分是在原答案的基礎(chǔ)上的改進雹洗，以求在方便程度和靈活性上有一定的提高香罐。
1. （翻譯）將 MATLAB 的 .mat 文件轉(zhuǎn)化成 OpenCV 的 cv::Mat 矩陣
由于 OpenCV 沒法直接讀取 MATLAB 的 .mat 文件，這就需要通過第三方的格式作為兩者之間的橋梁队伟⊙ù担考慮到 OpenCV 的 FileStorage 類提供了 XML/YML 文件的存取功能。
例如嗜侮，我們有一個如下的 yml 文件港令，文件名是 demo.yml
%YAML:1.0
Variable1: !!opencv-matrix
rows: 4
cols: 5
dt: f
data: [ -1.60522782e-03, -5.93489595e-03, 2.92204670e-03,
1.14785777e-02, -1.57432575e-02, -2.17529312e-02, 4.05947529e-02,
6.56594411e-02, 1.24527821e-02, 3.19751091e-02, 5.41692637e-02,
4.04683389e-02, 2.59191263e-03, 1.15112308e-03, 1.11024221e-02,
4.03668173e-03, -3.19138430e-02, -9.40114353e-03, 4.93452176e-02,
5.73473945e-02 ]
Variable2: !!opencv-matrix
rows: 7
cols: 2
dt: f
data: [ -2.17529312e-02, 4.05947529e-02, 5.73473945e-02,
6.56594411e-02, 1.24527821e-02, 3.19751091e-02, 5.41692637e-02,
4.03668173e-03, -3.19138430e-02, -9.40114353e-03, 4.93452176e-02,
4.04683389e-02, 2.59191263e-03, 1.15112308e-03 ]
接著，我們就可以用 OpenCV 的 FileStorage 類讀取儲存在 demo.yml 中的變量了锈颗，示例代碼如下：
#include <iostream>
#include <string>
#include <cv.h>
#include <highgui.h>
using namespace cv;
using namespace std;
int main (int argc, char * const argv[])
{
Mat var1;
Mat var2;
string demoFile = "demo.yml";
FileStorage fsDemo( demoFile, FileStorage::READ);
fsDemo["Variable1"] >> var1;
fsDemo["Variable2"] >> var2;
cout << "Print the contents of var1:" << endl;
cout << var1 << endl << endl;
cout << "Print the contents of var2:" << endl;
cout << var2 << endl;
fsDemo.release();
return 0;
}
至此顷霹，你所需要做的就是寫一個你自己的 Matlab parser，類似于如下這樣的：
function matlab2opencv( variable, fileName, flag)
[rows cols] = size(variable);
% Beware of Matlab's linear indexing
variable = variable';
% Write mode as default
if ( ~exist('flag','var') )
flag = 'w';
end
if ( ~exist(fileName,'file') || flag == 'w' )
% New file or write mode specified
file = fopen( fileName, 'w');
fprintf( file, '%%YAML:1.0\n');
else
% Append mode
file = fopen( fileName, 'a');
end
% Write variable header
fprintf( file, ' %s: !!opencv-matrix\n', inputname(1));
fprintf( file, ' rows: %d\n', rows);
fprintf( file, ' cols: %d\n', cols);
fprintf( file, ' dt: f\n');
fprintf( file, ' data: [ ');
% Write variable data
for i=1:rows*cols
fprintf( file, '%.6f', variable(i));
if (i == rows*cols), break, end
fprintf( file, ', ');
if mod(i+1,4) == 0
fprintf( file, '\n ');
end
end
fprintf( file, ']\n');
fclose(file);
由此击吱，我們就可以將 Matlab 變量儲存成 yml 文件淋淀，如下所示：
varA = rand( 3, 6);
varB = rand( 7, 2);
matlab2opencv( varA, 'newStorageFile.yml');
matlab2opencv( varB, 'newStorageFile.yml', 'a'); % append mode passed by 'a' flag
由此，我們就可以得到 newStorageFile.yml覆醇，如下所示：
%YAML:1.0
varA: !!opencv-matrix
rows: 3
cols: 6
dt: f
data: [ 0.430207, 0.979748, 0.258065,
0.262212, 0.221747, 0.318778, 0.184816,
0.438870, 0.408720, 0.602843, 0.117418,
0.424167, 0.904881, 0.111119, 0.594896,
0.711216, 0.296676, 0.507858]
varB: !!opencv-matrix
rows: 7
cols: 2
dt: f
data: [ 0.085516, 0.578525, 0.262482,
0.237284, 0.801015, 0.458849, 0.029220,
0.963089, 0.928854, 0.546806, 0.730331,
0.521136, 0.488609, 0.231594]
讀取 varA 和 varB 的方法就跟之前讀取 Variable1 和 Variable2 那樣朵纷。
2. 改進的代碼
需求
- 根據(jù)輸入的數(shù)據(jù)類型炭臭，自適應(yīng)地調(diào)整寫入到 yml 文件的數(shù)據(jù)類型，以方便后面
cv::Mat類型的數(shù)據(jù)訪問袍辞。
先來一個稍微改動一下鞋仍，寫入 int 類型的代碼：
function matlab2opencv_int( variable, fileName, flag)
[rows cols] = size(variable);
% Beware of Matlab's linear indexing
variable = variable';
% Write mode as default
if ( ~exist('flag','var') )
flag = 'w';
end
if ( ~exist(fileName,'file') || flag == 'w' )
% New file or write mode specified
file = fopen( fileName, 'w');
fprintf( file, '%%YAML:1.0\n');
else
% Append mode
file = fopen( fileName, 'a');
end
% Write variable header
fprintf( file, ' %s: !!opencv-matrix\n', inputname(1));
fprintf( file, ' rows: %d\n', rows);
fprintf( file, ' cols: %d\n', cols);
fprintf( file, ' dt: i\n');
fprintf( file, ' data: [ ');
% Write variable data
for i=1:rows*cols
fprintf( file, '%i', variable(i));
if (i == rows*cols), break, end
fprintf( file, ', ');
if mod(i+1,8) == 0
fprintf( file, '\n ');
end
end
fprintf( file, ']\n');
fclose(file);
有了上面的基礎(chǔ)，下面自動的代碼就可以是：
function dym_matlab2opencv(variable, fileName, flag, varClass)
% varClass: the variable class waiting for write:
% 'i': for int
% 'f': for float
% flag : Write mode
% 'w': for write
% 'a': for append
[rows, cols] = size(variable);
% Beware of Matlab's linear indexing
variable = variable';
% Write mode as default
if ( ~exist('flag','var') )
flag = 'w';
end
% float as default
if ( ~exist('varClass','var') )
if isfloat(variable)
varClass = 'f';
else isinteger(variable)
varClass = 'i';
end
end
if ( ~exist(fileName,'file') || flag == 'w' )
% New file or write mode specified
file = fopen( fileName, 'w');
fprintf( file, '%%YAML:1.0\n');
else
% Append mode
file = fopen( fileName, 'a');
end
% Write variable header
fprintf( file, ' %s: !!opencv-matrix\n', inputname(1));
fprintf( file, ' rows: %d\n', rows);
fprintf( file, ' cols: %d\n', cols);
fprintf( file, [' dt: ' varClass '\n']);
fprintf( file, ' data: [ ');
% Write variable data
for i=1:rows*cols
fprintf( file, ['%' varClass], variable(i));
if (i == rows*cols), break, end
fprintf( file, ', ');
if mod(i+1,8) == 0
fprintf( file, '\n ');
end
end
fprintf( file, ']\n');
fclose(file);
以上搅吁，初寫于 2015-05-02威创，更新于 2015-07-23。
首發(fā)于 Yimian Dai's Homepage谎懦，轉(zhuǎn)載請注明出處肚豺。
