medicaltorch

收录理由

医学影像深度学习项目里，真正耗时的往往不是模型结构，而是数据准备工作：读 NIFTI 文件、做强度归一化、把数据组织成 PyTorch 训练循环能直接吃进去的样子。MedicalTorch 把这些琐碎工作封装成现成的加载器、预处理函数和数据集类，团队做起分割或分类流程时，就不必再重复造轮子。项目自带一个脊髓灰质分割示例，还有一个 Colab 教程，完整跑通一遍端到端流程，对想先看看整体数据流再动手处理自己影像的人来说很实用。它严格遵循 PyTorch 原生的 Dataset 和 DataLoader 接口，因此能直接嵌入现有训练代码，不需要引入额外框架。对学术团队和小型原型项目，这比从零手写 NIFTI 解析器省心不少，是个更顺手的起点。