OpenCV cv::Mat set if

Question

是否有一种简单的方法将cv::Mat中的所有值设置为给定值，如果它们满足某些条件。例如，我有CV_32FC1，我想将所有值设置为0到20。在MATLAB中，我会简单地这样做:

M(M == 0) = 20;

Answer 1

可以使用

cv::Mat mask = M == 0;
M.setTo(0.5, mask);

然而，它包括使用额外的内存来创建掩码，但是使用opencv API是一个解决方案，因此可以应用于所有矩阵类型。如果考虑性能问题，可以直接引用Mat::data来针对具体矩阵类型优化此解决方案。

Answer 2

这是查找表的一个经典案例。它快速，简单，可以同时重新映射多个值。

Answer 3

感谢@marol的评论，我决定采用下面的实现。我使用c++ 11 lambda函数来条件哪些值需要更改。为了演示它的功能，我的条件是当值超出[MIN_VAL, MAX_VAL]范围时设置为DEFAULT_VAL:

#include <functional>
#define MatType float
#define MatCmpFunc std::function<bool(const MatType&)>
.
.
.
// function which accepts lambda function to condition values which need to
// be changed
void MatSetIf(cv::Mat& inputmat, const MatType& newval, MatCmpFunc func) {
  float* pmat = (float*)inputmat.data;
  // iterate and set only values which fulfill the criteria
  for (int idx = 0; idx < inputmat.total(); ++idx) {
    if (func(pmat[idx])) {
      pmat[idx] = newval;
    }
  }
}
.
.
.
void main() {
  cv::Mat mymat(100,100,CV_32FC1);
  const float MIN_VAL = 10;
  const float MAX_VAL = 1000;
  const float DEFAULT_VAL = -1;
  .
  .
  .
  // declare lambda function which returns true when mat value out of range
  MatCmpFunc func = [&](const DepthMatType& val) -> bool {
    return (val < MIN_VAL || val > MAX_VAL) ? true : false;
  };
  // use lambda func above to set all out of range values to 50
  Mat32FSetIf(mymat, DEFAULT_VAL, func);
  .
  .
  .
}