今天在学习 CUDA-NMS 算法时,遇到一个小问题,就是希望对一个二维数组进行排序,具体是这样的:
需要对存储目标框的二维数组进行排序,排序的方式是按照目标框的得分score从高往低排序,也就是结果数组的第二维的最后一个元素(dets[i][4])
sort()
直接进行排序,无法编译通过刚开始听人说,可以用sort()
直接进行排序,不过试了一下好像无法编译通过,示例代码:CPP_sort
#include // for std::sortconst int N = 3;
int arr[N][N] = {{3, 1, 7},{4, 1, 9},{5, 3, 2},};int main()
{// Initialize the array with some valuesfor (int i = 0; i < N; ++i)for (int j = 0; j < N; ++j)arr[i][j] = i * N + j;// Sort the array using a custom comparator functionstd::sort(arr, arr + N, [](const int a[], const int b[]) {return a[N-1] > b[N-1];});// The array is now sorted in descending order by the last element of the second dimensionreturn 0;
}
我个人感觉这里面比较本质的原因是:二维数组(的一行)无法像一维数组的元素一样,在排序的时候进行swap交换,所以在编译的时候会报错;
我个人觉得比较看起来有关联的一句报错信息如下所示:
报错信息表示:error: invalid array assignment,很可能就是表示无法直接使用dets[i]的方式直接赋值数组的一行;
代码示例:CPP_qsort_2d_array
Note:
这份代码示例没有用lambda表达式来书写,会使得代码的篇幅较长,之后会计划使用lambda表达式来进行改写。
上一篇:C与C++如何互相调用
下一篇:【C语言】重要函数atoi的使用