TorchMorph: CUDA-accelerated Morphological Transforms

intcomp · hf · 2026-08-26

TorchMorph is a PyTorch extension providing GPU-accelerated morphological and distance-transform operators across up to eight dimensions. It features a SciPy-compatible API to accelerate computational image processing tasks.

Original post →

More from Infra

Infra channel →