TorchIO-project/torchio

Medical imaging processing for AI applications.

What it solves

TorchIO 簡化了為深度學習準備 3D 醫療影像的過程。它解決了高效讀取、前處理、抽樣與增強大型 3D 醫療資料集的挑戰，這些資料集通常需要標準電腦視覺函式庫無法提供的專門轉換。

How it works

作為 PyTorch 的 Python 套件構建，TorchIO 提供了一套處理 3D 醫療影像的工具。它實作了通用的電腦視覺操作（如隨機仿射變換）以及模擬真實醫學成像偽影的領域特定轉換，例如 MRI 磁場不均勻與 k 空間運動偽影。它還包含一個 Queue 系統，以促進基於 patch 的訓練，從而實現更高效的資料抽樣。

Who it’s for

使用 PyTorch 建構 3D 醫療影像深度學習應用的研究人員與開發者。

Highlights

  • 3D Medical Image Support: 專用於讀取、前處理與寫入 3D 醫療資料的工具。
  • Domain-Specific Augmentation: 模擬 MRI 特有的偽影，如偏置場、鬼影與尖峰。
  • Patch-Based Sampling: 包含 Queue 系統，以最佳化訓練時影像 patch 的抽樣。
  • PyTorch Integration: 設計上與 PyTorch 生態系統無縫結合。