c++ - 如何使函数可以在 OpenCV 中处理不同类型的图像

Question

十字邮寄here

---

我已经构建了两个具有不同名称的函数来从不同的 Mat 对象中删除特定行，这是代码:

Mat drop_rows_int(Mat mat, vector<int> v) {
    Mat mat_new = Mat::zeros(mat.rows - v.size(), mat.cols, CV_32SC1);
    for (int i = 0, j = 0; i < mat.rows; i++) {
        if (find(v.begin(), v.end(), i) != v.end())
        {
            continue;
        }
        else
        {
            int*pmat = mat.ptr<int>(i);
            int*pmat_new = mat_new.ptr<int>(j);
            for (int w = 0; w < mat.cols; w++) {
                pmat_new[w] = pmat[w];
            }
            j++;
        }
    }
    return mat_new;
}

Mat drop_rows_uchar(Mat mat, vector<int> v) {
    Mat mat_new = Mat::zeros(mat.rows - v.size(), mat.cols, CV_8UC1);
    for (int i = 0, j = 0; i < mat.rows; i++) {
        if (find(v.begin(), v.end(), i) != v.end())
        {
            continue;
        }
        else
        {
            uchar*pmat = mat.ptr<uchar>(i);
            uchar*pmat_new = mat_new.ptr<uchar>(j);
            for (int w = 0; w < mat.cols; w++) {
                pmat_new[w] = pmat[w];
            }
            j++;
        }
    }
    return mat_new;
}

然后我可以在我的main() 函数中使用它

int main()
{
    Mat mat_uchar = (Mat_<uchar>(5, 4) << 5, 6, 0, 4, 0, 1, 9, 9, 100, 3, 5, 8, 200, 33, 1, 4, 8, 88, 23, 6);
    Mat new_mat_uchar = drop_rows_uchar(mat_uchar, {2,4});

    Mat mat_int = (Mat_<int>(5, 4) << 5, 6, 0, 4, 0, 1, 9, 9, 100, 3, 5, 8, 200, 33, 1, 4, 8, 88, 23, 6);
    Mat new_mat_int = drop_rows_int(mat_int, { 2,4 });

        return 0;
}

是的，我做到了。但是据我所知，Mat可以有7种深度，比如CV_8U, CV_8S, CV_16U, CV_16S、CV_32S、CV_32F 和 CV_64F，所以我必须构建 7 个不同名称的函数来做这样的事情？？谁能告诉我如何使用一个函数来实现它？？

Answer 1

你不能用 cv::Mat 做到这一点。但是，您可以使用 cv::Mat_ 并进行一些模板化:

template<typename T>
cv::Mat_<T> drop_rows_int(cv::Mat_ mat, vector<int> v) {
...
}

在这里你提取了 T 类型的指针。

只是一条建议，为了提高效率，我建议尽可能将 vector v 作为 const 引用发送。

---

这里是完整的解决方案:

#include "opencv/cv.h"
#include <vector>
#include <iostream>

template<typename T>
cv::Mat_<T> drop_rows(cv::Mat_<T> mat, const std::vector<int> &v) {
    cv::Mat_<T> mat_new = cv::Mat_<T>::zeros(mat.rows - v.size(), mat.cols);
    for (int i = 0, j = 0; i < mat.rows; i++) {
        if (find(v.begin(), v.end(), i) != v.end())
            continue;
        else {
            for (int w = 0; w < mat.cols; w++) {
                mat_new(j, w) = mat(i, w);
            }
            j++;
        }
    }
    return mat_new;
}

int main() {
    cv::Mat_<uchar> mat = (cv::Mat_<uchar>(5, 4) << 5, 6, 0, 4, 0, 1, 9, 9, 100, 3, 5, 8, 200, 1, 2, 3, 4, 5, 6, 7);
    auto result = drop_rows(mat, {2, 4});

    std::cout << mat << std::endl;;
    std::cout << result << std::endl;;

    return 0;
}

请注意，它仅适用于 Mat_，不适用于 Mat。