将高维矩阵从C ++传递给R

Question

我想将M \次N \次K矩阵从C ++传递给R.基本上我有一个cv::Mat对象，我想在R中有效地传递它。将M \次N cv::Mat转换为arma::Mat或者不是很难Rcpp::NumericMatrix。但如果我能得到更高维矩阵的包装器，那将会很棒。

Answer 1

现在您可以使用CRAN的xtensor包。

// [[Rcpp::plugins(cpp14)]]
// [[Rcpp::plugins(opencv)]]
// [[Rcpp::depends(xtensor)]]

#include <xtensor/xadapt.hpp>
#include <xtensor-r/rtensor.hpp>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <Rcpp.h>

using xtensor3d = xt::xtensor<double, 3>;
using rtensor3d = xt::rtensor<double, 3>;

template <typename T, int NCH, typename XT=xt::xtensor<T,3,xt::layout_type::column_major>>
XT to_xt(const cv::Mat_<cv::Vec<T, NCH>>& src) {
    std::vector<int> shape = {src.rows, src.cols, NCH};
    size_t size = src.total() * NCH;
    XT res = xt::adapt((T*) src.data, size, xt::no_ownership(), shape);
    return res;
}

// [[Rcpp::export]]
rtensor3d read_image(std::string file) {
    cv::Mat img = cv::imread(file);
    img.convertTo(img, CV_32FC3);
    xtensor3d res = img.channels() == 3 ? to_xt<double, 3>(img) : to_xt<double, 1>(img);
    return res;
}

opencv Rcpp插件应该导出为：

Rcpp::registerPlugin("opencv", function() {
    pkg_config_name <- "opencv4"
    pkg_config_bin <- Sys.which("pkg-config")
    checkmate::assert_file_exists(pkg_config_bin, access = "x")
    list(env = list(
        PKG_CXXFLAGS = system(paste(pkg_config_bin, "--cflags", pkg_config_name), intern = TRUE),
        PKG_LIBS = system(paste(pkg_config_bin, "--libs", pkg_config_name), intern = TRUE)
    ))
})