我可以使用 opencv 函数(如 cv::multiply 和 std::vector<cv::Mat> 作为输入和输出参数吗?



阅读opencv的文档,在我看来,我可以使用std::vector作为任何函数的输入和输出参数,接收InputArray并输出OutputArray类,例如cv::multiply

http://docs.opencv.org/master/d4/d32/classcv_1_1__InputArray.html#gsc.tab=0

http://docs.opencv.org/master/d2/de8/group__core__array.html#ga979d898a58d7f61c53003e162e7ad89f

但是,我似乎无法让它工作:

float data[3][3] = {{1, 2, 3}, {4, 5, 6}, {7, 8, 9}};
cv::Mat mat = cv::Mat(3, 3, CV_32F, data);
std::vector<cv::Mat> vecMat = {mat, mat, mat};

std::vector<cv::Mat> outputVecMat;
cv::multiply(vecMat, vecMat, outputVecMat);

我在这里做错了什么还是只是错过了解释文档?

InputArray基本上是一个围绕cv::Matstd::vector<T>的包装类。但是multiply(除法相同(期望一个 1 维或 2 维矩阵。例如,有效代码可以是:

#include <opencv2/opencv.hpp>
#include <vector>
int main()
{
    float data[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
    cv::Mat mat = cv::Mat(3, 3, CV_32F, data);
    // Multiply cv::Mat
    cv::Mat outMat;
    cv::multiply(mat, mat, outMat);
    // Multiply std::vector
    std::vector<float> vec(data, data + 9);
    std::vector<float> outVec;
    cv::multiply(vec, vec, outVec);
    return 0;
}

如果你想将矩阵乘以一个std::vector<cv::Mat>(有 3 维(,你需要做一些额外的工作,定义你自己的函数(mymul这里(:

#include <opencv2/opencv.hpp>
#include <vector>

void mymul(const std::vector<cv::Mat>& src1, const std::vector<cv::Mat>& src2, std::vector<cv::Mat>& dst)
{
    dst.clear();
    // Be sure that src1 and src2 have the same number of elements
    CV_Assert(src1.size() == src2.size());
    dst.resize(src1.size());
    for (int i = 0; i < src1.size(); ++i)
    {
        cv::multiply(src1[i], src2[i], dst[i]);
    }
}
int main()
{
    float data[] = { 1, 2, 3, 4, 5, 6, 7, 8, 9 };
    cv::Mat mat = cv::Mat(3, 3, CV_32F, data);
    std::vector<cv::Mat> vecMat = { mat, mat, mat };
    std::vector<cv::Mat> outputVecMat;
    // Call custom multiply that works on std::vector<cv::Mat>
    mymul(vecMat, vecMat, outputVecMat);
    return 0;
}

这里的问题不是InputArray没有正确形成。实际上,它是正确创建的,kind等于STD_VECTOR_MAT。问题是multiply期望InputArraycv::Matstd::vector<T>,其中T是内置类型(intfloat等(。

您可以看到接受std::vector<cv::Mat>的OpenCV函数具有类型为cv::InputArrayOfArrayscv::OutputArrayOfArrays的参数,其中显然std::vector的类型Tcv::Mat类型。

最新更新