torchio: 도메인 특화 아티팩트를 포함한 3D 의료 영상의 전처리 및 증강을 위한 PyTorch 기반 툴킷

해결하는 문제

TorchIO는 딥러닝을 위한 3D 의료 영상 준비 과정을 단순화합니다. 이는 표준 컴퓨터 비전 라이브러리가 제공하지 않는 특수한 변환이 자주 필요한 대규모 3D 의료 데이터셋을 효율적으로 읽고, 전처리하고, 샘플링하고, 증강하는 문제를 해결합니다.

작동 방식

PyTorch를 위한 Python 패키지로 구축된 TorchIO는 3D 의료 영상을 처리하기 위한 도구 모음을 제공합니다. 일반적인 컴퓨터 비전 작업(예: random affine transformations)과 MRI magnetic field inhomogeneity 및 k-space motion artifacts와 같은 실제 의료 영상 아티팩트를 시뮬레이션하는 도메인 특화 변환을 모두 구현합니다. 또한 패치 기반 학습을 용이하게 하는 Queue 시스템을 포함하여 데이터의 더 효율적인 샘플링을 가능하게 합니다.

대상 사용자

PyTorch를 사용하여 3D 의료 영상용 딥러닝 애플리케이션을 구축하는 연구자 및 개발자.

주요 특징

• 3D 의료 영상 지원: 3D 의료 데이터를 읽고, 전처리하고, 쓰는 데 특화된 도구.
• 도메인 특화 증강: bias fields, ghosting, spikes와 같은 MRI 특화 아티팩트를 시뮬레이션.
• 패치 기반 샘플링: 학습을 위한 이미지 패치 샘플링을 최적화하기 위한 Queue 시스템 포함.
• PyTorch 통합: PyTorch 생태계 내에서 원활하게 작동하도록 설계됨.