TorchMorph 是一个新的 PyTorch 扩展,它将形态学变换带到 GPU 上,解决了现有仅 CPU 实现(如 SciPy)的局限性。它提供了 22 个 CUDA 加速算子,用于各种形态学类型和距离变换,支持多达八个空间维度。这带来了显著的性能提升,TorchMorph 在灰度形态学方面的吞吐量比 SciPy 高出 1100 倍,在距离变换方面高出 350 倍,同时保持了高精度。 AI
影响 通过启用 GPU 加速的形态学操作,加速了 AI 管道中的图像处理任务。
排序理由 该条目描述了一个新的软件库及其性能基准,属于研究类别。[lever_c_demoted from research: ic=1 ai=1.0]
在 Hugging Face Daily Papers 阅读 →
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →