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 avatar

qubvel/segmentation_models

0
View on GitHub↗

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

Segmentation Models

This is an image segmentation framework and masking toolkit for constructing binary and multi-class neural network architectures. It serves as a deep learning encoder wrapper that integrates pre-trained convolutional neural network architectures into semantic segmentation models.

The library enables the use of pre-trained backbones to isolate complex patterns and leverages transfer learning to accelerate training. It provides a collection of overlap-based loss functions and precision metrics specifically designed to evaluate and refine the accuracy of image masks.

The toolkit covers the full segmentation pipeline, including image input normalization, the assembly of encoder-decoder architectures, and the calculation of performance scores using overlap and precision metrics.

Features

  • Image Segmentation - Provides a framework for constructing neural networks that isolate specific objects within images.
  • PyTorch Semantic Segmentation Libraries - Provides a comprehensive framework for building semantic segmentation models using pre-trained encoders and specialized loss functions.
  • Deep Learning Architectures - Provides an interface for integrating pre-trained CNN architectures into semantic segmentation models.
  • Overlap-Based - Provides Tversky-based loss functions to handle class imbalance between target objects and background pixels.
  • Mask Overlap Optimization - Applies specialized loss functions to refine the overlap of predicted masks against ground truth data.
  • Backbone Model Integration - Integrates pre-trained convolutional backbones to identify and isolate complex patterns in image data.
  • Vision Loss Functions - Implements specialized vision loss functions that measure mask overlap to improve model accuracy.
  • Transfer Learning - Leverages pre-trained weights from image classification networks to accelerate training of segmentation models.
  • Semantic Segmentation Architectures - Implements encoder-decoder architectures specifically for pixel-wise semantic segmentation.
  • Computer Vision Toolkits - Offers a toolkit of overlap-based loss functions and precision metrics for evaluating segmentation accuracy.
  • Feature Map Upsamplers - Increases spatial resolution of feature maps using transposed convolutions to align predictions with original image size.
  • Image Data Preprocessing - Prepares raw image data to ensure compatibility between data sources and model encoders.
  • Instance Segmentation Loss Calculators - Computes specialized loss functions to determine the exact accuracy of generated image masks.
  • Segmentation Model Abstractions - Wraps deep learning layers into high-level API classes for rapid segmentation network assembly.
  • Segmentation Evaluation Metrics - Calculates overlap and precision metrics to evaluate how well the model identifies object boundaries.
  • Image Preprocessing Pipelines - Includes workflows for normalizing image pixel values for compatibility with chosen encoders.
  • Input Normalizers - Provides utilities for standardizing image pixel values to match pre-trained encoder distributions.
4,917 stars·1,047 forks·Python·mit·18 views

Star history

Star history chart for qubvel/segmentation_modelsStar history chart for qubvel/segmentation_models

How this analysis was created: This summary and feature list are AI-generated from collected project material and can contain mistakes. Stars, license and language are imported from GitHub. Inclusion does not mean that we have tested or audited this project. Check the source documentation for any feature you depend on. Learn more on our About page.

Projects sharing features with Segmentation Models

These projects share indexed features with Segmentation Models. Shared tags can include platform or build tooling; verify the primary use case before treating a result as a replacement.
  • nvlabs/segformerNVlabs avatar

    NVlabs/SegFormer

    3,347View on GitHub↗

    SegFormer is a semantic segmentation framework and transformer-based model designed for pixel-level image classification. It provides a deep learning architecture that assigns class labels to pixels using a hierarchical transformer encoder and a multi-layer perceptron decoder. The framework utilizes a hierarchical transformer encoder to process multi-scale features through a pyramid of blocks and an all-MLP decoder to aggregate these features without complex attention mechanisms. It incorporates overlap patch embedding to preserve local continuity and sequential self-attention reduction to ma

    Pythonade20kcityscapessemantic-segmentation
    View on GitHub↗3,347
  • 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
  • nyandwi/machine_learning_completeNyandwi avatar

    Nyandwi/machine_learning_complete

    4,983View on GitHub↗

    This is an interactive notebook-based course that teaches machine learning from Python fundamentals through deep learning and natural language processing. It uses real datasets and multiple frameworks within a structured, hands-on curriculum that combines concise explanations with executable code cells, built-in datasets, and embedded exercise checkpoints. Learning progresses through data preparation and exploration, classical machine learning workflows, computer vision with convolutional neural networks, and natural language processing with deep learning, all delivered as a cohesive progressi

    Jupyter Notebookcomputer-visiondata-analysisdata-science
    View on GitHub↗4,983
  • tingsongyu/pytorch_tutorialTingsongYu avatar

    TingsongYu/PyTorch_Tutorial

    8,018View on GitHub↗

    This project is a comprehensive collection of educational examples and reference implementations for building vision and language models using PyTorch. It serves as a deep learning tutorial covering the end-to-end process of developing neural networks, from initial architecture definition to final production deployment. The repository provides detailed guides on implementing a wide range of domain-specific models, including convolutional neural networks for object detection and segmentation, as well as transformer and recurrent architectures for natural language processing. It emphasizes gene

    Python
    View on GitHub↗8,018
Compare all 30 related projects→

Frequently asked questions

What does qubvel/segmentation_models do?

This is an image segmentation framework and masking toolkit for constructing binary and multi-class neural network architectures. It serves as a deep learning encoder wrapper that integrates pre-trained convolutional neural network architectures into semantic segmentation models.

What are the main features of qubvel/segmentation_models?

The main features of qubvel/segmentation_models are: Image Segmentation, PyTorch Semantic Segmentation Libraries, Deep Learning Architectures, Overlap-Based, Mask Overlap Optimization, Backbone Model Integration, Vision Loss Functions, Transfer Learning.

Which projects share features with qubvel/segmentation_models?

Projects with overlapping indexed features include: nvlabs/segformer — SegFormer is a semantic segmentation framework and transformer-based model designed for pixel-level image… zhengpeng7/birefnet — BiRefNet is a PyTorch image segmentation framework designed for high-precision binary mask generation. It functions as… nyandwi/machine_learning_complete — This is an interactive notebook-based course that teaches machine learning from Python fundamentals through deep… tingsongyu/pytorch_tutorial — This project is a comprehensive collection of educational examples and reference implementations for building vision… mic-dkfz/nnunet — nnU-Net is a PyTorch-based deep learning framework for the supervised semantic segmentation of 2D and 3D biomedical… xuebinqin/u-2-net — U-2-Net is a PyTorch image segmentation framework and computer vision saliency model designed to generate…