Hi all,
I have written torchio 410, a Python package with tools for patch-based training and inference of 3D medical images and multiple transforms for data augmentation typically used in the field. The torchio.transforms 56 API is similar to torchvision.transforms. All the processing is done using PyTorch, NumPy and ITK 4.

  
      
      GitHub
  
  
    

fepegar/torchio 410

Tools for loading, augmenting and writing 3D medical images on PyTorch. - fepegar/torchio


  
  
    
    
  
  


The library is still very immature, so contributions and feedback are very welcome  If you have the opportunity to try it out, please let me know what you think and report any issues on GitHub or here.