torchio: 一个基于 PyTorch 的工具包，用于通过领域特定伪影对 3D 医学图像进行预处理和增强

它解决了什么问题

TorchIO 简化了为深度学习准备 3D 医学图像的过程。它解决了高效读取、预处理、采样和增强大型 3D 医学数据集的挑战，这些数据集通常需要标准计算机视觉库无法提供的专门转换。

它是如何工作的

作为 PyTorch 的 Python 包构建，TorchIO 提供了一套处理 3D 医学图像的工具。它实现了通用的计算机视觉操作（如随机仿射变换）和模拟真实世界医学成像伪影的领域特定转换，例如 MRI 磁场不均匀性和 k-space 运动伪影。它还包含一个 Queue 系统以促进基于 patch 的训练，从而实现更高效的数据采样。

它是为谁准备的

使用 PyTorch 构建 3D 医学成像深度学习应用的研究人员和开发人员。

亮点

• 3D 医学图像支持：用于读取、预处理和写入 3D 医学数据的专门工具。
• 领域特定增强：模拟 MRI 特有的伪影，如偏置场、重影和尖峰。
• 基于 Patch 的采样：包含一个 Queue 系统以优化训练中图像 patch 的采样。
• PyTorch 集成：旨在在 PyTorch 生态系统中无缝工作。