Armadillo与OpenCV矩阵相互转换方法

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);（待验证）

参考链接：Armadillo与opencv库之间调用