#327 · Primary category: Computer Vision

medicaltorch

computer-vision deep-learning machine-learning medical-imaging python pytorch

A medical imaging framework for Pytorch

Project last updated:04/26/24

GitHub Stars

872

Forks

129

Contributors

8

License

Apache-2.0

Why we included this project

Most of the work in a medical imaging deep learning project is data plumbing, not model design: reading NIFTI files, normalizing intensities, and building datasets that play nicely with PyTorch's training loop. MedicalTorch packages that grunt work into ready-made loaders, pre-processors, and dataset classes, so a team starting a segmentation or classification pipeline can skip re-writing that glue. The repo ships with a spinal cord gray matter segmentation example and a Colab tutorial that walks through a real end-to-end run, which helps when you want to see the full flow before wiring up your own volumes. Because it sticks to PyTorch's native Dataset and DataLoader conventions, it drops into an existing training loop without forcing a new stack on you. For academic groups and small teams prototyping clinical vision models, it is a cleaner starting point than hand-writing a NIFTI parser from scratch.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category