awesome-repositories.com
Blog
MCP
awesome-repositories.com

Discover the best open-source repositories with AI-powered search.

ExploreCurated searchesOpen-source alternativesSelf-hosted softwareBlogSitemap
ProjectMCP serverAboutHow we rankPress
LegalPrivacyTerms
© 2026 Bringes Technology SRL·VAT RO45896025·hello@awesome-repositories.com
qubvel-org avatar

qubvel-org/segmentation_models.pytorch

0
View on GitHub↗
11,622 stars·1,837 forks·Python·MIT·21 viewssmp.readthedocs.io↗

Segmentation Models.pytorch

This is a PyTorch semantic segmentation library designed for building image masking frameworks. It provides a collection of over 500 pretrained convolutional and transformer-based encoders and various decoder architectures to perform binary and multiclass pixel-level classification.

The library features a modular backbone integration that decouples encoder choice from decoder logic. It supports custom input channel configurations and encoder depth tuning, allowing the modification of input layers to accept non-standard channel counts while preserving pretrained weights. Some configurations also allow for the attachment of auxiliary classification heads to produce both a segmentation mask and a global image label.

Additional capabilities include preprocessing functions aligned with pretrained encoder weights and tools for exporting trained models to the ONNX format for cross-platform deployment. The system also supports integration with model hubs for saving and loading weights.

Features

  • PyTorch Semantic Segmentation Libraries - Provides a comprehensive library for building image segmentation models using PyTorch with a vast collection of pretrained encoders.
  • Semantic Segmentation - Provides a comprehensive library for building semantic segmentation models using pretrained encoders and various decoders.
  • Binary Segmentations - Enables identifying a single object class from the background by assigning a binary label to each pixel.
  • Multiclass Segmentations - Provides the capability to categorize every pixel in an image into one of several distinct object classes.
  • Encoder-Decoder Architectures - Implements a variety of encoder-decoder architectures to map latent representations to pixel-level masks.
  • Modular Backbone Architectures - Features a modular backbone integration that allows swapping between convolutional and transformer-based encoders.
  • Pretrained Encoder Collections - Offers a collection of over 500 pretrained convolutional and transformer-based encoders for feature extraction.
  • Image Masking Frameworks - Implements a system for performing binary and multiclass pixel-level classification to create object masks.
  • Computer Vision Workflows - Facilitates custom vision workflows by allowing adjustments to input channels and encoder depths.
  • ONNX Model Exporters - Includes tools for serializing PyTorch computational graphs into the universal ONNX format for cross-platform deployment.
  • Data Preprocessing - Ships preprocessing functions that normalize and transform input images to align with pretrained encoder weights.
  • Model Depth Optimization - Enables adjusting the number of downsampling operations in the encoder to balance model complexity and inference speed.
  • Preprocessing Pipelines - Provides preprocessing functions aligned with specific pretrained encoder weights to ensure consistent normalization.
  • Adaptive Input Layers - Allows modifying the first layer of pretrained encoders to accept custom input channel counts while preserving weights.
  • Input Channel Adaptations - Allows configuring the number of input channels to support tensors with arbitrary dimensions while preserving pretrained weights.

Star history

Star history chart for qubvel-org/segmentation_models.pytorchStar history chart for qubvel-org/segmentation_models.pytorch

How this analysis was created: This summary and feature list were written by an AI model that read the project's README and public documentation pages. Each feature links to the documentation it came from; stars, license and language come straight from the GitHub API. The model does not read the source code, and the analysis is refreshed when the project is re-analysed. Learn more on our About page.

AI search

Explore more awesome repositories

Describe what you need in plain English — the AI ranks thousands of curated open-source projects by relevance.

Start searching with AI

Open-source alternatives to Segmentation Models.pytorch

Similar open-source projects, ranked by how many features they share with Segmentation Models.pytorch.
  • zhengpeng7/birefnetZhengPeng7 avatar

    ZhengPeng7/BiRefNet

    3,173View on GitHub↗

    BiRefNet is a PyTorch image segmentation framework designed for high-precision binary mask generation. It functions as a bilateral image segmentation model used to isolate foreground objects from complex backgrounds, as well as a specialized tool for camouflaged object detection and industrial defect detection. The project is designed for export to the ONNX format, which facilitates cross-platform deployment and inference. It supports custom model fine-tuning on user-provided image and mask datasets to adapt the model for specialized professional use cases. The system covers high-resolution

    Pythonbackground-removalbirefnetcamouflaged-object-detection
    View on GitHub↗3,173
  • tingsongyu/pytorch-tutorial-2ndTingsongYu avatar

    TingsongYu/PyTorch-Tutorial-2nd

    4,555View on GitHub↗

    This project is a comprehensive instructional resource and course for building neural networks using PyTorch. It covers the fundamental building blocks of deep learning, including tensor manipulation, automatic differentiation, and the construction of modular neural network components. The repository serves as a technical guide for several specialized domains. It provides implementation details for computer vision tasks such as image classification, object detection, and semantic segmentation, as well as natural language processing workflows involving transformers, recurrent networks, and gen

    Jupyter Notebookcomputer-visiondeepsortdiffusion-models
    View on GitHub↗4,555
  • autogluon/autogluonautogluon avatar

    autogluon/autogluon

    9,997View on GitHub↗

    AutoGluon is an automated machine learning framework and multimodal library designed to automate the end-to-end pipeline from data preprocessing to high-accuracy model training and validation. It functions as an automated model trainer for tabular, image, text, and time series data, as well as a tool for time series forecasting and foundation model finetuning. The project is distinguished by its ability to jointly process and fuse different data types, allowing for the construction of multimodal neural networks that integrate images, text, and structured tables. It supports zero-shot inferenc

    Pythonautogluonautomated-machine-learningautoml
    View on GitHub↗9,997
  • facebookresearch/dinov3facebookresearch avatar

    facebookresearch/dinov3

    9,613View on GitHub↗

    This project is a self-supervised vision foundation model based on a vision transformer architecture. It is designed to learn dense visual representations from unlabeled images, serving as a general-purpose backbone for a wide variety of downstream vision tasks. The system is distinguished by its use of self-distillation and masked image modeling to extract semantic and geometric features. It also incorporates an image-text alignment model that maps visual embeddings to textual descriptions, enabling zero-shot image recognition, zero-shot segmentation, and cross-modal retrieval. The project

    Jupyter Notebook
    View on GitHub↗9,613
See all 30 alternatives to Segmentation Models.pytorch→

Frequently asked questions

What does qubvel-org/segmentation_models.pytorch do?

This is a PyTorch semantic segmentation library designed for building image masking frameworks. It provides a collection of over 500 pretrained convolutional and transformer-based encoders and various decoder architectures to perform binary and multiclass pixel-level classification.

What are the main features of qubvel-org/segmentation_models.pytorch?

The main features of qubvel-org/segmentation_models.pytorch are: PyTorch Semantic Segmentation Libraries, Semantic Segmentation, Binary Segmentations, Multiclass Segmentations, Encoder-Decoder Architectures, Modular Backbone Architectures, Pretrained Encoder Collections, Image Masking Frameworks.

What are some open-source alternatives to qubvel-org/segmentation_models.pytorch?

Open-source alternatives to qubvel-org/segmentation_models.pytorch include: zhengpeng7/birefnet — BiRefNet is a PyTorch image segmentation framework designed for high-precision binary mask generation. It functions as… tingsongyu/pytorch-tutorial-2nd — This project is a comprehensive instructional resource and course for building neural networks using PyTorch. It… autogluon/autogluon — AutoGluon is an automated machine learning framework and multimodal library designed to automate the end-to-end… facebookresearch/dinov3 — This project is a self-supervised vision foundation model based on a vision transformer architecture. It is designed… milesial/pytorch-unet — Pytorch-UNet is a deep learning implementation designed for semantic image segmentation. It provides a framework for… apple/corenet — Corenet is a deep learning training framework and computer vision model library designed for developing neural…