挺呆的一个操作,仍然是python 借助 ctypes调用一个C++(其实是C)的实现……
就不上传资源了,直接把代码贴出来算了。
#include <math.h>
#include <stdlib.h>
#include <stdio.h>
#include <float.h>
#define EPSILON 1e-6
void find_local_maximas(float* matrix, unsigned char* result, int startRow, int endRow, int startCol, int endCol, int M, int N);
void find_local_maximas(float* matrix, unsigned char* result, int startRow, int endRow, int startCol, int endCol, int M, int N) {
if (startRow >= endRow - 1 || startCol >= endCol - 1) {
return;
}
int midRow = (startRow + endRow) / 2;
int midCol = (startCol + endCol) / 2;
// 查找左上象限的局部最大值
find_local_maximas(matrix, result, startRow, midRow, startCol, midCol, M, N);
// 查找右上象限的局部最大值
find_local_maximas(matrix, result, startRow, midRow, midCol, endCol, M, N);
// 查找左下象限的局部最大值
find_local_maximas(matrix, result, midRow, endRow, startCol, midCol, M, N);
// 查找右下象限的局部最大值
find_local_maximas(matrix, result, midRow, endRow, midCol, endCol, M, N);
// 处理交界处
for (int i = startRow; i < midRow; ++i) {
for (int j = startCol; j < midCol; ++j) {
float value = matrix[(i * N) + j];
if (fabs(value - 1.0) < EPSILON) {
result[(i * N) + j] = 1; // 如果值接近1.0,直接标记为1
} else {
float top = (i == 0) ? -INFINITY : matrix[((i - 1) * N) + j];
float bottom = (i == M - 1) ? -INFINITY : matrix[((i + 1) * N) + j];
float left = (j == 0) ? -INFINITY : matrix[(i * N) + (j - 1)];
float right = (j == N - 1) ? -INFINITY : matrix[(i * N) + (j + 1)];
if (value > top && value > bottom && value > left && value > right) {
result[(i * N) + j] = 1;
} else {
result[(i * N) + j] = 0;
}
}
}
}
}
// 外部接口函数
extern "C" {
void find_local_maximas_wrapper(float* matrix, unsigned char* result, int M, int N) {
find_local_maximas(matrix, result, 0, M, 0, N, M, N);
}
}
总之传入一个np.float32的矩阵matrix,一个np.zeros((M,N),dtype = np.uint8)的接受传出的矩阵result,再把MN传进去就齐活了。注意matrix需要把最大值设为1,不然传不进去……
我们这个实现里本身是要求如果局部最大值是全局最大值就也标记为局部最大值,所以加了一个判定接不接近1的控制,不想加的回头改改就行。
怎么在python里调用可以看我另一篇文章和代码,总之比较简单我就不贴代码了。