Segmentation-models-pytorch

Latest version: v0.3.4

Safety actively analyzes 682404 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

Page 1 of 3

0.3.4

Updates
- 🤗 Hugging Face integration: you can save, load, and share models with HF [Hub](https://huggingface.co/models), see [example notebook](https://github.com/qubvel-org/segmentation_models.pytorch/blob/main/examples/save_load_model_and_share_with_hf_hub.ipynb).

Full log
* To support albumentations >= 1.4.0 some functions need to be renamed by CallShaul in https://github.com/qubvel-org/segmentation_models.pytorch/pull/870
* Updated false positve and false negative rate functions in functional.py by vermavinay982 in https://github.com/qubvel-org/segmentation_models.pytorch/pull/855
* Add HF hub mixin by qubvel in https://github.com/qubvel-org/segmentation_models.pytorch/pull/876
* use precommit for code linting by Borda in https://github.com/qubvel-org/segmentation_models.pytorch/pull/829
* Add Ruff for formatting and linting by qubvel in https://github.com/qubvel-org/segmentation_models.pytorch/pull/877
* Add docs config by qubvel in https://github.com/qubvel-org/segmentation_models.pytorch/pull/878
* Update docs by qubvel in https://github.com/qubvel-org/segmentation_models.pytorch/pull/879
* Add `create_model` to docs by qubvel in https://github.com/qubvel-org/segmentation_models.pytorch/pull/883
* Update ruff to version 0.5.2 and workflows update by Smartappli in https://github.com/qubvel-org/segmentation_models.pytorch/pull/892
* Fix hub_mixin.py pop error by ytzfhqs in https://github.com/qubvel-org/segmentation_models.pytorch/pull/909
* Update HF mixin by qubvel in https://github.com/qubvel-org/segmentation_models.pytorch/pull/910

New Contributors
* CallShaul made their first contribution in https://github.com/qubvel-org/segmentation_models.pytorch/pull/870
* vermavinay982 made their first contribution in https://github.com/qubvel-org/segmentation_models.pytorch/pull/855
* Borda made their first contribution in https://github.com/qubvel-org/segmentation_models.pytorch/pull/829
* Smartappli made their first contribution in https://github.com/qubvel-org/segmentation_models.pytorch/pull/892
* ytzfhqs made their first contribution in https://github.com/qubvel-org/segmentation_models.pytorch/pull/909

**Full Changelog**: https://github.com/qubvel-org/segmentation_models.pytorch/compare/v0.3.3...v0.3.4

0.3.3

Updates
- Pytorch image models (timm) version upgrade to 0.9.2

0.3.2

Updates
- Added Apple's Mobile One encoder from [repo](https://github.com/apple/ml-mobileone) (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

0.3.1

Updates
- Added Mix Vision Transformer encoder from SegFormer [[official code](https://github.com/NVlabs/SegFormer)] [[paper](https://arxiv.org/abs/2105.15203)]. Use argument `encoder_name="mit_b0"` (or mit_b1..b5) to create a model.
- Minor typo fixes and docs updates

0.3.0

Updates
- Added `smp.metrics` module with different metrics based on confusion matrix, see [docs](https://smp.readthedocs.io/en/latest/metrics.html)
- Added new [notebook](https://github.com/qubvel/segmentation_models.pytorch/blob/master/examples/binary_segmentation_intro.ipynb) with training example using pytorch-lightning ![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)
- 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!

0.2.1

Updates
- Universal timm encoder. 400+ pretrained encoders from [timm](https://github.com/rwightman/pytorch-image-models) available with `tu-` prefix. List of available encoders [here](https://smp.readthedocs.io/en/latest/encoders_timm.html).
- Minor fixes and improvements.

Page 1 of 3

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.