opencv - 如何将 remove_if 与 vector<point2f> 一起使用

Question

我有一个向量，其中有很多我想删除的 x,y 位置的 NaN(做一些 opencv 工作)。我不知道如何使用 remove_if 删除 NaN(与删除一起使用时)。如果向量是浮点数或整数但不是point2f，我已经看到了很多例子。任何简单的例子都会很有帮助。谢谢。

Answer 1

您可以使用 lambda 函数、仿函数或函数指针。这是一个带有 lambda 函数的示例:

#include <opencv2/opencv.hpp>
#include <algorithm>
#include <iostream>
#include <cmath>

using namespace cv;
using namespace std;

int main(int argc, char ** argv)
{
    vector<Point2f> pts{ Point2f(1.f, 2.f), Point2f(3.f, sqrt(-1.0f)), Point2f(2.f, 3.f) };

    cout << "Before" << endl;
    for (const auto& p : pts) {
        cout << p << " ";
    }
    cout << endl;

    pts.erase(remove_if(pts.begin(), pts.end(), [](const Point2f& p)
    {
        // Check if a coordinate is NaN
        return isnan(p.x) || isnan(p.y);
    }), pts.end());

    cout << "After" << endl;
    for (const auto& p : pts) {
        cout << p << " ";
    }
    cout << endl;

    return 0;
}

这将打印:

Before
[1, 2] [3, -1.#IND] [2, 3]
After
[1, 2] [2, 3]