We present TorchIO, an open-source Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images for deep learning. It follows the design of PyTorch and relies on standard medical image processing libraries such as SimpleITK or NiBabel to efficiently process large 3D images during the training of convolutional neural networks. We provide multiple generic as well as magnetic-resonance-imaging-specific operations for preprocessing and augmentation of medical images. TorchIO is an open-source project with code, comprehensive examples and extensive documentation shared at https://github.com/fepegar/torchio.
翻译:我们介绍Torchio,这是一个开放源码的Python图书馆,用于高效装载、预处理、扩增和补丁抽样医疗图象,供深层学习,它遵循PyTorch的设计,并依靠标准的医疗图象处理图书馆,如SimplicITK或NiBabel,在培训进化神经网络期间高效处理大型的3D图象,我们为医疗图象的预处理和扩增提供多种通用的和磁共振成像操作,Torchio是一个开源项目,有代码、全面实例和广泛文件,可在https://github.com/fepegar/tochio上查阅。