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 • 0.2B • Updated Apr 12 • 8

smp-hub/upernet-swin-small

Image Segmentation • 0.1B • Updated Apr 12 • 4

smp-hub/upernet-swin-tiny

Image Segmentation • 0.1B • Updated Apr 12 • 135

smp-hub/upernet-convnext-xlarge

Image Segmentation • 0.4B • Updated Apr 12 • 72

smp-hub/upernet-convnext-large

Image Segmentation • 0.2B • Updated Apr 12 • 35

smp-hub/upernet-convnext-base

Image Segmentation • 0.1B • Updated Apr 12 • 12

smp-hub/upernet-convnext-small

Image Segmentation • 0.1B • Updated Apr 12 • 5

smp-hub/upernet-convnext-tiny

Image Segmentation • 0.1B • Updated Apr 12 • 112

smp-hub/dpt-large-ade20k

Image Segmentation • 0.3B • Updated Apr 6 • 316

smp-hub/mobileone_s4.imagenet

Image Classification • 0.0B • Updated Jan 15 • 113

View 140 models

datasets 0

None public yet