Segmentation Models Pytorch

non-profit

https://github.com/qubvel-org/segmentation_models.pytorch

qubvel-org

AI & ML interests

None defined yet.

Organization Card

Community About org cards

Python library with Neural Networks for Image
Segmentation based on PyTorch.

📚 Project Documentation 📚
GitHub

Collections 3

View 3 collections

models 140

smp-hub/upernet-swin-large

Image Segmentation • Updated Apr 12, 2025 • 9

smp-hub/upernet-swin-small

Image Segmentation • Updated Apr 12, 2025 • 3

smp-hub/upernet-swin-tiny

Image Segmentation • Updated Apr 12, 2025 • 25

smp-hub/upernet-convnext-xlarge

Image Segmentation • Updated Apr 12, 2025 • 2

smp-hub/upernet-convnext-large

Image Segmentation • Updated Apr 12, 2025 • 1

smp-hub/upernet-convnext-base

Image Segmentation • Updated Apr 12, 2025 • 2

smp-hub/upernet-convnext-small

Image Segmentation • Updated Apr 12, 2025 • 51

smp-hub/upernet-convnext-tiny

Image Segmentation • Updated Apr 12, 2025 • 10

smp-hub/dpt-large-ade20k

Image Segmentation • Updated Apr 6, 2025 • 50 • 1

smp-hub/mobileone_s4.imagenet

Image Classification • Updated Jan 15, 2025 • 31

View 140 models

datasets 0

None public yet