Skip to content

Segmentation Models - v0.3.0

Compare
Choose a tag to compare
@qubvel qubvel released this 29 Jul 10:32

Updates

  • Added smp.metrics module with different metrics based on confusion matrix, see docs
  • Added new notebook with training example using pytorch-lightning Open In Colab
  • Improved handling of incorrect input image size error (checking image size is 2^n)
  • Codebase refactoring and style checks (black, flake8)
  • Minor typo fixes and bug fixes

Breaking changes

  • utils module is going to be deprecated, if you still need it import it manually from segmentation_models_pytorch import utils

Thanks a lot for all contributors!