Skip to content

mlf-core prediction package for U-Net models trained on the LiTS dataset.

License

Notifications You must be signed in to change notification settings

luiskuhn/liver-ct-segmentation-package

Repository files navigation

liver-ct-segmentation-package

Github Workflow Build liver-ct-segmentation-package Status Github Workflow Tests Status PyPI Status Documentation Status Dependabot Enabled

Prediction package for reproducible U-Net models trained on the LiTS dataset. These models are trained using the mlf-core framework (https://github.com/mlf-core/liver-ct-segmentation/).

Features

  • Model download CLI: liver-ct-seg-model-dl
  • Prediction CLI: liver-ct-seg-pred
  • Uncertainty of prediction CLI: liver-ct-seg-uncert
  • Input feature importance (Guided Grad-CAM) CLI: liver-ct-seg-feat-ggcam

Credits

This package was created with mlf-core using cookiecutter.

About

mlf-core prediction package for U-Net models trained on the LiTS dataset.

Resources

License

Stars

Watchers

Forks

Packages

No packages published