GPU peak detection with torch

czbiohub-sf / shrimPy

shrimPy: Smart High-throughput Robust Imaging & Measurement in Python

BSD 3-Clause "New" or "Revised" License

7 stars 1 forks source link

GPU peak detection with torch #129

Closed ziw-liu closed 5 months ago

ziw-liu commented 5 months ago

Re-implemented an approximate version of skimage.feature.peak_local_max with torch to run on GPU. On the mantis workstation (P4000 with 8GB VRAM) it takes ~4s to compute.

ziw-liu commented 5 months ago

Most of the runtime is spent on copying the ~3 GB test image through PCIe...