torchio

收录理由

TorchIO 把原始医学扫描数据和 PyTorch 训练循环之间的数据准备环节打通了。它能读取常见的神经影像格式，用强度变换和空间变换做数据增强，还能把很大的三维体数据切成小块，这样分割或分类网络就能在整份扫描无法一次性装进内存的情况下正常训练。这些变换是以 PyTorch 风格组合起来的，预处理自然就留在训练管道内部，不用散落到一堆独立脚本里。做 MRI、CT 或其他体扫描的团队，用它来统一数据准备、增加真实的变化，省去了自己手写加载器的麻烦。而且这个代码库本身也是一个很好的范例，展示了现代医学影像处理管道是怎么一步步搭建起来的。