Skip to content
/ GaNDLF Public
forked from mlcommons/GaNDLF

A generalizable application framework for segmentation, regression, and classification using PyTorch

License

Notifications You must be signed in to change notification settings

slyalin/GaNDLF

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GaNDLF

Codacy
Code style: black

The Generally Nuanced Deep Learning Framework for segmentation, regression and classification.

GaNDLF all options

Why use this?

  • Supports multiple
    • Deep Learning model architectures
    • Data dimensions (2D/3D)
    • Channels/images/sequences
    • Prediction classes
    • Domain modalities (i.e., Radiology Scans and Digitized Histopathology Tissue Sections)
    • Problem types (segmentation, regression, classification)
    • Multi-GPU (on same machine) training
  • Built-in
    • Nested cross-validation (and related combined statistics)
    • Support for parallel HPC-based computing
    • Support for training check-pointing
    • Support for Automatic mixed precision
  • Robust data augmentation, courtesy of TorchIO
  • Handles imbalanced classes (e.g., very small tumor in large organ)
  • Leverages robust open source software
  • No need to write any code to generate robust models

Citation

Please cite the following article for GaNDLF (full PDF):

@misc{pati2021gandlf,
      title={GaNDLF: A Generally Nuanced Deep Learning Framework for Scalable End-to-End Clinical Workflows in Medical Imaging}, 
      author={Sarthak Pati and Siddhesh P. Thakur and Megh Bhalerao and Spyridon Thermos and Ujjwal Baid and Karol Gotkowski and Camila Gonzalez and Orhun Guley and Ibrahim Ethem Hamamci and Sezgin Er and Caleb Grenko and Brandon Edwards and Micah Sheller and Jose Agraz and Bhakti Baheti and Vishnu Bashyam and Parth Sharma and Babak Haghighi and Aimilia Gastounioti and Mark Bergman and Anirban Mukhopadhyay and Sotirios A. Tsaftaris and Bjoern Menze and Despina Kontos and Christos Davatzikos and Spyridon Bakas},
      year={2021},
      eprint={2103.01006},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

Documentation

GaNDLF has extensive documentation and it is arranged in the following manner:

Contributing

Please see the contributing guide for more information.

Disclaimer

  • The software has been designed for research purposes only and has neither been reviewed nor approved for clinical use by the Food and Drug Administration (FDA) or by any other federal/state agency.
  • This code (excluding dependent libraries) is governed by the Apache License, Version 2.0 provided in the LICENSE file unless otherwise specified.

Contact

For more information or any support, please post on the Discussions section.

About

A generalizable application framework for segmentation, regression, and classification using PyTorch

Resources

License

Code of conduct

Security policy

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.4%
  • Roff 0.6%