MATLAB sort 函数的 OpenCV 实现

在 MATLAB 里，返回排序后的矩阵以及对应原矩阵的索引是在 sort 一个函数搞定，但在 OpenCV 中，其功能分别被分配到了 cv::sort 和 cv::sortIdx 里，cv::sort 负责返回排序后的矩阵，cv::sortIdx 负责返回对应原矩阵的索引。
还有在 MATLAB 里，1 和 2 用来分别指示是对列还是对行进行排序，'ascend' 和 'descend' 用来指示是升序还是降序。在 OpenCV 中，我们用类似于 CV_SORT_EVERY_ROW + CV_SORT_ASCENDING 这样的方式来一并指定对列还是对行以及升序还是降序，其指示值定义如下，所以可以组合出 4 种不同的方式：

#define CV_SORT_EVERY_ROW    0
#define CV_SORT_EVERY_COLUMN 1
#define CV_SORT_ASCENDING    0
#define CV_SORT_DESCENDING   16

• CV_SORT_EVERY_ROW + CV_SORT_ASCENDING：对矩阵的每行按照升序排序；
• CV_SORT_EVERY_ROW + CV_SORT_DESCENDING：对矩阵的每行按照降序排序；
• CV_SORT_EVERY_COLUMN + CV_SORT_ASCENDING：对矩阵的每列按照升序排序；
• CV_SORT_EVERY_COLUMN + CV_SORT_DESCENDING：对矩阵的每列按照降序排序；

具体代码可见 OpenCV-and-CPP-for-MATLAB-Users Demo_OpenCVFunctions 文件夹下的 demo_sort_sortIdx.cpp 函数，为了方便起见，将代码陈列如下：

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

void demo_sort_sortIdx()
{
    int testArrLen = 5;
    cv::Mat_<int> testArr = cv::Mat::zeros(2, testArrLen, CV_32S);
    testArr(0,0) = 87;
    testArr(0,1) = 65;
    testArr(0,2) = 98;
    testArr(0,3) = 12;
    testArr(0,4) = 55;
    testArr(1,0) = 86;
    testArr(1,1) = 66;
    testArr(1,2) = 97;
    testArr(1,3) = 17;
    testArr(1,4) = 54;

    cv::Mat_<int> sortArr, sortIdxArr;
    cv::sort(testArr, sortArr, CV_SORT_EVERY_ROW  +  CV_SORT_ASCENDING);
    cv::sortIdx(testArr, sortIdxArr, CV_SORT_EVERY_ROW  +  CV_SORT_ASCENDING);

    std::cout<<"testArr = "<<testArr<<std::endl;
    std::cout<<"sortArr = "<<sortArr<<std::endl;
    std::cout<<"sortIdxArr = "<<sortIdxArr<<std::endl;
}

本文是开源项目 OpenCV-and-CPP-for-MATLAB-Users 文档的一部分，更多的文章和完整代码请移步项目的 Github主页。

转载请注明出处。