SOURCE-LINKED INTELLIGENCE

TorchMorph: CUDA-accelerated Morphological Transforms

arXiv · AI, language, vision and robotics · article · Aug 25, 2026 · UTC

Morphological transforms are long-standing tools for shape and mask processing, but the de facto reference implementation in the Python ecosystem, i.e. scipy.ndimage, is CPU-only, single-array, and therefore unusable inside a GPU training loop without an expensive device-to-host round trip. GPU vision libraries built on PyTorch cover a narrow subset of these operators, typically restricted to two spatial dimensions and flat structuring elements. We present TorchMorph, a lightweight PyTorch extension that closes this gap. TorchMorph exposes 22 public operators covering binary morphology, greysc

Read original source ↗ Open in workspace

recordType

paper

region

Global