Segmentation Models - v0.3.4

Updates

• 🤗 Hugging Face integration: you can save, load, and share models with HF Hub, see example notebook.

Full log

• To support albumentations >= 1.4.0 some functions need to be renamed by @CallShaul in #870
• Updated false positve and false negative rate functions in functional.py by @vermavinay982 in #855
• Add HF hub mixin by @qubvel in #876
• use precommit for code linting by @Borda in #829
• Add Ruff for formatting and linting by @qubvel in #877
• Add docs config by @qubvel in #878
• Update docs by @qubvel in #879
• Add create_model to docs by @qubvel in #883
• Update ruff to version 0.5.2 and workflows update by @Smartappli in #892
• Fix hub_mixin.py pop error by @ytzfhqs in #909
• Update HF mixin by @qubvel in #910

New Contributors

• @CallShaul made their first contribution in #870
• @vermavinay982 made their first contribution in #855
• @Borda made their first contribution in #829
• @Smartappli made their first contribution in #892
• @ytzfhqs made their first contribution in #909

Full Changelog: v0.3.3...v0.3.4

Segmentation Models - v0.3.3

Updates

• Pytorch image models (timm) version upgrade to 0.9.2

Segmentation Models - v0.3.2

Updates

• Added Apple's Mobile One encoder from repo (use encoder_name="mobileone_s{0..4}").
• Pytorch image models (timm) version upgrade to 0.6.12 (500+ encoders available)
• Minor typo fixes and docs updates

Breaking changes

• Minimum Python version 3.6 -> 3.7

Thanks @VadimLevin, @kevinpl07, @Abd-elr4hman

Segmentation Models - v0.3.1

Updates

• Added Mix Vision Transformer encoder from SegFormer [official code] [paper]. Use argument encoder_name="mit_b0" (or mit_b1..b5) to create a model.
• Minor typo fixes and docs updates

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 manually from segmentation_models_pytorch import utils

Thanks a lot for all contributors!

Segmentation Models - v0.2.1

Updates

• Universal timm encoder. 400+ pretrained encoders from timm available with tu- prefix. List of available encoders here.
• Minor fixes and improvements.

Segmentation Models - v0.2.0

Updates

• New architecture: MANet (#310)
• New encoders from timm: mobilenetv3 (#355) and gernet (#344)
• New loss functions in smp.losses module (smp.utils.losses would be deprecated in future versions)
• New pretrained weight initialization for first convolution if in_channels > 3
• Updated timm version (0.4.12)
• Bug fixes and docs improvement

Thanks to @azkalot1 @JulienMaille @originlake @Kupchanski @loopdigga96 @zurk @nmerty @ludics @Vozf @markson14 and others!

Segmentation Models - v0.1.3

Updates

• New architecture Unet++ (#279)
• New encoders RegNet, ResNest, SK-Net, Res2Net (#286)
• Updated timm version (0.3.2)
• Improved docstrings and typehints for models
• Project documentation on https://smp.readthedocs.io

Thanks to @azkalot1 for the new encoders and architecture!

Segmentation Models - v0.1.2

Fixes

• Fix pytorch-efficientnet package version in requirements.txt to strict 0.6.3 (#260)

Segmentation Models - v0.1.1

Updates

• New decoders DeepLabV3, DeepLabV3+, PAN
• New backbones (encoders) timm-efficientnet*
• New pretrained weights (ssl, wsl) for resnets
• New pretrained weights (advprop) for efficientnets

And some small fixes.

Thanks @IlyaDobrynin @gavrin-s @lizmisha @suitre77 @thisisiron @phamquiluan and all other contributers!