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

smp-hub/upernet-swin-small

Image Segmentation • Updated Apr 12 • 26

smp-hub/upernet-swin-tiny

Image Segmentation • Updated Apr 12 • 33

smp-hub/upernet-convnext-xlarge

Image Segmentation • Updated Apr 12 • 14

smp-hub/upernet-convnext-large

Image Segmentation • Updated Apr 12 • 15

smp-hub/upernet-convnext-base

Image Segmentation • Updated Apr 12 • 22

smp-hub/upernet-convnext-small

Image Segmentation • Updated Apr 12 • 235

smp-hub/upernet-convnext-tiny

Image Segmentation • Updated Apr 12 • 62

smp-hub/dpt-large-ade20k

Image Segmentation • Updated Apr 6 • 143 • 1

smp-hub/mobileone_s4.imagenet

Image Classification • Updated Jan 15 • 32

View 140 models

datasets 0

None public yet