在OpenCV中cv :: Mat的自定義類型

Question

我想用我的自定義類型創建一個使用cv :: Mat的數組。這裏是我的代碼：

typedef struct{ 
    int x; 
    int y; 
    float prob; 
}CellXY; 

void main(){ 

    cv::Mat_<CellXY> map(10, 10); 

    std::cout << "Width = " << map.cols << ", Height = " << map.rows << std::endl; 
    std::cout << "### Assignment ###" << std::endl; 
    for (int j=0; j<map.rows; j++){ 
    for (int i=0; i<map.cols; i++){ 
     map.at<CellXY>(j, i).x = i; 
     map.at<CellXY>(j, i).y = j; 
     map.at<CellXY>(j, i).prob = (float)(i + j * map.cols); 

     std::cout << map.at<CellXY>(j, i).prob 
     << " (" << map.at<CellXY>(j, i).x 
     << "," << map.at<CellXY>(j, i).y 
     << ")\t"; 
    } 
    std::cout << std::endl; 
    } 

    std::cout << "### Display ###" << std::endl; 
    for (int j=0; j<map.rows; j++){ 
    for (int i=0; i<map.cols; i++){ 
     std::cout << map.at<CellXY>(j, i).prob 
     << " (" << map.at<CellXY>(j, i).x 
     << "," << map.at<CellXY>(j, i).y 
     << ")\t"; 
    } 
    std::cout << std::endl; 
    } 
    map.release(); 
} 

「分配」和「顯示」部分之間。然而結果是不一樣的：

如何使這些結果一樣嗎？

Answer 1

這是因爲OpenCV不知道你班上的DataType。你需要一個專門爲它，像這樣：

typedef struct { 
    int x; 
    int y; 
    float prob; 
}CellXY; 


// Add this!!!!!! 
namespace cv { 
template<> class DataType<CellXY> 
{ 
public: 
    typedef CellXY  value_type; 
    typedef value_type work_type; 
    typedef value_type channel_type; 
    typedef value_type vec_type; 
    enum { generic_type = 0, 
     depth  = CV_32F, // Might cause problems... I'm saying it's all made of floats (and not 2 int a 1 float) 
     channels  = 3, 
     fmt   = (int)'p', 
     type   = CV_MAKETYPE(depth, channels) 
    }; 
}; 
} 

void main() { 
    ... 
} 

但是，在一般情況下，我認爲這是更好地避免這些執行細節搞亂了，並且只使用一個更好的數據結構。建議爲：僅對原始類型使用Mat。事實上，你不能在其他OpenCV函數無論如何使用它...

這裏有一些方法來解決你的問題：

1. 使用另一種數據結構：例如一個std::vector<std::vector<CellXY>>或「展開」std::vector<CellXY>

2. 使類CellXYMatrix與存儲CellXY涉及：

   class CellXYMatrix 
   { 
       std::vector<CellXY> _mat; // your matrix of CellXY 
       int _r; // rows 
       int _c; // cols 
   
   public: 
       CellXYMatrix(int rows, int cols) : _r(rows), _c(cols), _mat(rows * cols) {}; 
   
       void set(int row, int col, const CellXY& cell) { 
        _mat[row * _c + col] = cell; 
       } 
       CellXY& get(int row, int col) { 
        return _mat[row * _c + col]; 
       } 
   } 
   

   最後，您可以重載operator()做出類似的OpenCV墊的訪問。

3. 如果在CellXY中的x和y字段指向矩陣位置，爲什麼需要它們？您可以簡單地爲您的概率設置Mat1f（又名Mat_<float>），x，y是矩陣中的位置。