如何与Eigen进行元素比较?

时间:2018-04-24 19:13:47

标签: c++ opencv eigen

我尝试使用Eigen在C ++中实现以下伪代码:

img_binary = +1*(img>img_mean) + -1*(img<img_mean)

即。我试图将灰度图像转换为二进制图像,使得大于图像平均值的值为+1且小于图像平均值为-1。到目前为止,我有以下内容:

cv::Mat cv_image
cv_image = cv::imread(img_path, CV_LOAD_IMAGE_GRAYSCALE)

MatrixXf eig_image;
cv::cv2eigen(cv_image, eig_image):

float image_mean = eig_image.mean();

ArrayXXf bin_image;
bin_image = eig_image.array() > image_mean;

我在最后一行收到错误,说我混合了不同的数字类型。关于如何与Eigen进行元素比较的任何建议?

1 个答案:

答案 0 :(得分:2)

Eigen中最简单的解决方案是

ArrayXXf bin_image = (eig_image.array() > image_mean)
    .select(ArrayXXf::Constant(1.0f,eig_image.rows(),eig_image.cols()), -1.0f);

另一种选择是:

ArrayXXf::Constant

不幸的是,必须使用.select作为一个参数,因为没有[]方法接受两个标量值

但是,除非您打算在Eigen中进行进一步处理,否则您应该考虑使用相应的OpenCV方法threshold