segmentation-models-pytorch has 823K monthly downloads and 12K GitHub stars. Version 0.5.0 supports Python 3.9+, first published in 2019.
It is actively maintained under a permissive license, last released in Apr 2025.
It is built on top of huggingface-hub, numpy and pillow. It is used by 77 tracked packages, including torchgeo, inference-models and gandlf.
GitHub topics include segmentation, image-processing and pspnet.
Package Insights
- No releases in 440 days despite active commits
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- huggingface-hub >=0.24
- numpy >=1.19.3
- pillow >=8
- safetensors >=0.3.1
- timm >=0.9
- torch >=1.8
- torchvision >=0.9
- tqdm >=4.42.1
12 optional dependencies
- autodocsumm[docs]
- gitpython[test]
- huggingface-hub[docs]
- packaging[test]
- pytest[test]
- pytest-cov[test]
- pytest-xdist[test]
- ruff[test]
- setuptools[test]
- six[docs]
- sphinx[docs]
- sphinx-book-theme[docs]
Used By
- torchgeo
- inference-models
- gandlf
- terratorch
- omnicloudmask
- satcube
- arcjetcv
- datamint
- pytorch-segmentation-models-trainer
- inference-exp
- cloudsen12-models
- moonshine
- imsy-htc
- marss2l
- deepvisiontools
- medicalmultitaskmodeling
- celldetection
- models-package
- doc2graph
- roofai
- towbintools
- spacr
- quadra
- earthnet-models-pytorch
- cell-cycle-classification
- mirc-dataset-handler
- silicrop
- olmoearth-runner
- elm-prune
- deeptrees
- phenotyping-segmentation
- geowatch
- cnn-framework
- betaearth
- skema-kelp
- medsegbench
- earsegmentationai
- spacr-nightly
- fedcore
- volume-segmantics-vsui
- brainseg-ai
- sefixlines
- geo-neural-network
- torch-pixel-classifier
- river-segmentation
- miles-credit
- deepml
- wmh-seg
- olimp
- pathpretrain