Segmentation Models - v0.3.0
Updates
- Added
smp.metrics
module with different metrics based on confusion matrix, see docs - Added new notebook with training example using pytorch-lightning
- 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 manuallyfrom segmentation_models_pytorch import utils
Thanks a lot for all contributors!