HomeModelsSegmentation Models Pytorch

Segmentation Models Pytorch Models

45

45 segmentation-models-pytorch models on MLForge, ranked by downloads — including work from smp-hub, giswqs, DimitrisMantas, smp-test-models. Compare, download and run them locally on your own GPU. Free tier.

Start freeDesktop app

Top Segmentation Models Pytorch models

E smp-hub/efficientnet-b2.imagenet Image Classification 81.7K 🤗 HF E smp-hub/efficientnet-b3.imagenet Image Classification 66.4K 🤗 HF E smp-hub/efficientnet-b0.imagenet Image Classification 54.4K 1 🤗 HF R smp-hub/resnet18.imagenet Image Classification 46.3K 🤗 HF R smp-hub/resnet34.imagenet Image Classification 45.0K 🤗 HF M smp-hub/mobilenet_v2.imagenet Image Classification 39.0K 🤗 HF M smp-hub/mit_b1.imagenet Image Classification 37.6K 🤗 HF E smp-hub/efficientnet-b1.imagenet Image Classification 35.8K 🤗 HF R smp-hub/resnet50.imagenet Image Classification 18.2K 🤗 HF M smp-hub/mit_b3.imagenet Image Classification 16.1K 1 🤗 HF E smp-hub/efficientnet-b4.imagenet Image Classification 8.4K 🤗 HF M smp-hub/mit_b2.imagenet Image Classification 4.4K 🤗 HF E smp-hub/efficientnet-b5.imagenet Image Classification 4.2K 🤗 HF S smp-hub/se_resnext50_32x4d.imagenet Image Classification 2.7K 🤗 HF U smp-hub/upernet-swin-large Image Segmentation 999 🤗 HF S smp-hub/segformer-b2-512x512-ade-160k Image Segmentation 693 🤗 HF S smp-hub/segformer-b2-1024x1024-city-160k Image Segmentation 346 🤗 HF S smp-hub/segformer-b0-512x512-ade-160k Image Segmentation 341 🤗 HF W giswqs/whu-building-unetplusplus-efficientnet-b4 Image Segmentation 239 4 🤗 HF D smp-hub/dpt-large-ade20k Image Segmentation 188 1 🤗 HF R DimitrisMantas/RoofSense Image Segmentation ·12.4M params 166 🤗 HF S smp-hub/segformer-b5-640x640-ade-160k Image Segmentation 126 1 🤗 HF D smp-test-models/deeplabv3-tu-resnet18 Image Segmentation 115 1 🤗 HF D smp-test-models/deeplabv3plus-tu-resnet18 Image Segmentation 111 🤗 HF U glean4/unet_spinachcount_taylor_target_res_ft_from_crop_and_weed_20250926 Image Segmentation 104 🤗 HF U smp-test-models/unet-tu-resnet18 Image Segmentation 103 🤗 HF M smp-test-models/manet-tu-resnet18 Image Segmentation 96 🤗 HF U smp-test-models/unetplusplus-tu-resnet18 Image Segmentation 94 🤗 HF L smp-test-models/linknet-tu-resnet18 Image Segmentation 93 🤗 HF S smp-test-models/segformer-tu-resnet18 Image Segmentation 92 🤗 HF P smp-test-models/pspnet-tu-resnet18 Image Segmentation 90 🤗 HF U smp-test-models/upernet-tu-resnet18 Image Segmentation 90 🤗 HF F smp-test-models/fpn-tu-resnet18 Image Segmentation 89 🤗 HF P smp-test-models/pan-tu-resnet18 Image Segmentation 88 🤗 HF D smp-test-models/dpt-tu-test_vit Image Segmentation 87 🤗 HF S smp-hub/segformer-b5-1024x1024-city-160k Image Segmentation 79 🤗 HF U smp-hub/upernet-swin-tiny Image Segmentation 62 🤗 HF D rosadecsai/dt_unet-hyperdoc Image Segmentation ·24.5M params 57 🤗 HF S AnikiFan/single-channel-breast-segmentation-manet Image Segmentation ·31.8M params 53 🤗 HF U smp-hub/upernet-convnext-base Image Segmentation 41 🤗 HF S AnikiFan/single-channel-breast-segmentation-deeplabv3plus Image Segmentation ·22.5M params 25 🤗 HF S AnikiFan/single-channel-breast-segmentation-Unet Image Segmentation ·24.4M params 16 1 🤗 HF O qubvel-hf/oxford-pet-segmentation Image Segmentation 8 1 🤗 HF O SeppSepp/oxford-pet-segmentation Image Segmentation ·6.3M params 7 🤗 HF U Diamantis99/unetplusplus_se_resnext101_32x4d Image Segmentation ·72.5M params 4 🤗 HF

Other frameworks

Transformers 5659Diffusers 479Timm 341Ultralytics 225Transformers.Js 212Open Clip 209Sentence Transformers 197Pytorch 154Mlx 107PaddleOCR 97Ml Agents 93Stable Baselines3 90Peft 51Gguf 48Ggml 42Diffusion Single File 39Nemo 37Yolov5 36Mlx Audio 34Coreml 31