Run similar algorithm to 'peak_local_max' on GPU

Hi,

I am new in working on GPU and I using the library skimage.feature, focus on the function peak_local_max.
This algorithm find peaks in an image as coordinate list or boolean mask.
Peaks are the local maxima in a region of 2 * min_distance + 1 (i.e. peaks are separated by at least min_distance).
I was trying to find a way to run it on GPU without success.

Do you have any recommendation for similar algorithm that run on the GPU?

You may want to look at Rapids.ai CUCIM library: https://docs.rapids.ai/api/cucim/stable/api.html

function should be cucim.skimage.feature.peak_local_max()

1 Like