Armadillo与OpenCV矩阵相互转换方法
static void Cv_mat_to_arma_mat(const cv::Mat1b& cv_mat_in, arma::uchar_mat& arma_mat_out)
{//convert unsigned char cv::Mat to arma::Mat<uchar>
for (int r = 0; r < cv_mat_in.rows; r++) {
for (int c = 0; c < cv_mat_in.cols; c++) {
arma_mat_out(r, c) = cv_mat_in.data[r*cv_mat_in.cols + c];// / 255.0
}
}
};
template<typename T>
static void Arma_mat_to_cv_mat(const arma::Mat<T>& arma_mat_in, cv::Mat_<T>& cv_mat_out)
{
cv::transpose(cv::Mat_<T>(static_cast<int>(arma_mat_in.n_cols),
static_cast<unsigned char>(arma_mat_in.n_rows),
const_cast<T*>(arma_mat_in.memptr())),
cv_mat_out);
};
注意:
- 使用时要初始化arma mat。
- cv转arma:
Cv_mat_to_arma_mat(cvimg,armaimg)
;(已验证) - arma转cv:
Arma_mat_to_cv_mat<double>(armaimg,cvimg)
;(待验证)