在 OpenCV 中分配矩阵元素

Assign matrix element in OpenCV

本文关键字：元素 分配 OpenCV      更新时间：2023-10-16

在OpenCV中将元素从一个源矩阵分配给目标矩阵的最通用方法是什么？我总是发现自己编码这样的东西，这并不完全优雅：

cv::Mat source; // CV_32FC3
cv::Mat dest;   // CV_32FC3
// ...
switch( source.channels() )
{
case 1:
    {
        dest.at<float>( y, x ) = source.at<float>( j, i );
        break;
    }
case 2:
    {
        dest.at<cv::Vec2f>( y, x ) = source.at<cv::Vec2f>( j, i );
        break;
    }
case 3:
    {
        dest.at<cv::Vec3f>( y, x ) = source.at<cv::Vec3f>( j, i );
        break; 
    }
case 4:
    {
        dest.at<cv::Vec4f>( y, x ) = source.at<cv::Vec4f>( j, i );
        break;
    }
}

我想知道推荐的方法是什么，一定有一些通用的单行代码，对吧？

跨不同数据类型工作的解决方案的加分(例如，将 n 通道浮点元素分配给 n 通道双精度或短元素(

有一个用于cv::Vec<T, n>的模板化版本，因此您可以制作一个模板函数，采用源和目标类型，以及通道数，如下所示：

template<typename TSrc, typename TDst, int N>
void assign(const cv::Mat& src, cv::Mat& dst, int x, int y)
{
    assert(src.channels() == N);
    assert(dst.channels() == N);
    assert(src.size() == dst.size());
    const cv::Vec<TSrc, N>& srcVec = src.at<cv::Vec<TSrc, N>>(x, y);
    cv::Vec<TDst, N>& dstVec = dst.at<cv::Vec<TDst, N>>(x, y);
    for (int i = 0; i < N; ++i)
        dstVec[i] = static_cast<TDst>(srcVec[i]);
}
int main(int argc, char* argv[])
{   //sample usage:
    cv::Mat src(5, 5, CV_8UC3, cv::Scalar::all(255)), dst(5, 5, CV_32FC3, cv::Scalar::all(1));
    assign<unsigned char, float, 3>(src, dst, 3, 3);
}
相关文章：