TorchMorph 是一个新的 PyTorch 扩展，提供 GPU 加速的形态学和距离变换算子。它提供了 22 个算子，包括二值和灰度形态学，以及各种距离变换，这些算子都实现了为融合的 CUDA 内核。该库的设计旨在模仿 SciPy API，可以轻松集成到现有流程中，并提供比基于 CPU 的实现显著的速度提升。 AI
影响 加速 GPU 训练循环中的图像处理和掩码分析，可能加快计算机视觉领域 AI 模型开发的速度。
排序理由 该集群描述了一个新的软件库及其相关论文，详细介绍了其技术能力和性能改进。
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →
完整方法见我们的编辑标准。
TorchMorph is a PyTorch extension providing GPU-accelerated morphological and distance-transform operators across up to eight dimensions with a SciPy-compatible API.
arXiv:2608.24738v1 Announce Type: new Abstract: 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 tr…