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 • 4

smp-hub/upernet-swin-small

Image Segmentation • Updated Apr 12 • 23

smp-hub/upernet-swin-tiny

Image Segmentation • Updated Apr 12 • 100

smp-hub/upernet-convnext-xlarge

Image Segmentation • Updated Apr 12 • 9

smp-hub/upernet-convnext-large

Image Segmentation • Updated Apr 12 • 3

smp-hub/upernet-convnext-base

Image Segmentation • Updated Apr 12 • 216

smp-hub/upernet-convnext-small

Image Segmentation • Updated Apr 12 • 391

smp-hub/upernet-convnext-tiny

Image Segmentation • Updated Apr 12 • 569

smp-hub/dpt-large-ade20k

Image Segmentation • Updated Apr 6 • 130

smp-hub/mobileone_s4.imagenet

Image Classification • Updated Jan 15 • 74

View 140 models

datasets 0

None public yet