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
xuebinqin avatar

xuebinqin/U-2-Net

0
View on GitHub↗
9,773 stars·1,614 forks·Python·Apache-2.0·15 views

U 2 Net

U-2-Net is a PyTorch image segmentation framework and computer vision saliency model designed to generate high-resolution foreground-background masks. It functions as an AI background removal tool that identifies and isolates the most visually prominent objects within an image.

The model utilizes a nested U-structure design to detect salient objects, creating precise cutouts by predicting saliency maps. These capabilities enable the separation of main subjects from their surroundings to create transparent images.

The framework covers several image processing workflows, including automatic background removal, salient object detection, and human portrait extraction for isolating head and shoulder regions. It employs a symmetric encoder-decoder path and multi-scale feature aggregation to localize object boundaries.

Features

  • Background Removal Tools - Provides an AI-powered tool for isolating subjects from image backgrounds to create transparent cutouts.
  • Saliency Models - Implements a research-based model that detects visually significant regions for portrait extraction and masking.
  • Salient - Isolates visually prominent objects within an image to create a map for background removal.
  • Image Segmentation - Partitions images into distinct foreground and background regions for downstream computer vision tasks.
  • Salient - Uses a nested U-structure architecture to identify and isolate the most prominent objects in an image.
  • Nested U-Structures - Utilizes a nested U-structure design to extract deep features through repeated downsampling and upsampling paths.
  • PyTorch Implementations - Provides a complete model implementation using the PyTorch framework for image segmentation research.
  • PyTorch Semantic Segmentation Libraries - Implements a PyTorch framework for generating high-resolution binary masks for image segmentation.
  • Saliency-Driven Learning - Trains the network to produce probability maps that isolate the most visually prominent objects from the background.
  • Image Background Removal - Separates the main object from its surroundings to create transparent cutouts.
  • Symmetric Encoder-Decoders - Employs a symmetric encoder-decoder path to recover spatial resolution for precise object boundary localization.
  • Multi-Scale Feature Aggregation - Implements multi-scale feature aggregation to improve the accuracy of object masks across various sizes.
  • Portrait Masking - Extracts human head and shoulder regions using face detection and cropping to create high-quality masks.
  • Background removal - Listed in the “Background removal” section of the Ailia Models awesome list.

Star history

Star history chart for xuebinqin/u-2-netStar history chart for xuebinqin/u-2-net

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

Frequently asked questions

What does xuebinqin/u-2-net do?

U-2-Net is a PyTorch image segmentation framework and computer vision saliency model designed to generate high-resolution foreground-background masks. It functions as an AI background removal tool that identifies and isolates the most visually prominent objects within an image.

What are the main features of xuebinqin/u-2-net?

The main features of xuebinqin/u-2-net are: Background Removal Tools, Saliency Models, Salient, Image Segmentation, Nested U-Structures, PyTorch Implementations, PyTorch Semantic Segmentation Libraries, Saliency-Driven Learning.

What are some open-source alternatives to xuebinqin/u-2-net?

Open-source alternatives to xuebinqin/u-2-net include: danielgatis/rembg — Rembg is a machine learning-based toolkit designed for automated image background removal and subject segmentation. It… nvlabs/segformer — SegFormer is a semantic segmentation framework and transformer-based model designed for pixel-level image… qubvel/segmentation_models — This is an image segmentation framework and masking toolkit for constructing binary and multi-class neural network… peterl1n/robustvideomatting — RobustVideoMatting is a deep learning video matting tool and PyTorch library designed to remove backgrounds from… wzmiaomiao/deep-learning-for-image-processing — This project is a PyTorch-based computer vision library and deep learning image processing framework. It provides a… datawhalechina/thorough-pytorch — This project is an educational resource and comprehensive guide for implementing and deploying deep learning models…

Open-source alternatives to U 2 Net

Similar open-source projects, ranked by how many features they share with U 2 Net.
  • danielgatis/rembgdanielgatis avatar

    danielgatis/rembg

    21,911View on GitHub↗

    Rembg is a machine learning-based toolkit designed for automated image background removal and subject segmentation. It functions as a versatile engine that identifies and extracts subjects from images, supporting diverse input methods including individual files, directory-based batch processing, and live binary data streams. The project distinguishes itself through its flexible integration options, offering a command-line interface for local automation, a library for programmatic access, and an HTTP service for remote requests. It utilizes deep learning architectures to classify pixels and ge

    Pythonbackground-removalimage-processingpython
    View on GitHub↗21,911
  • 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
  • qubvel/segmentation_modelsqubvel avatar

    qubvel/segmentation_models

    4,917View on GitHub↗

    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

    Pythondensenetefficientnetfpn
    View on GitHub↗4,917
  • peterl1n/robustvideomattingPeterL1n avatar

    PeterL1n/RobustVideoMatting

    9,244View on GitHub↗

    RobustVideoMatting is a deep learning video matting tool and PyTorch library designed to remove backgrounds from videos and extract human subjects. It utilizes a temporal video segmentation model to ensure consistent matting and reduce flickering across video frames. The project includes a cross-platform model exporter that converts trained neural networks into various runtime formats. This allows for model deployment across multiple environments, including web and mobile applications. The framework provides capabilities for temporal video background removal and AI video post-production with

    Pythonaicomputer-visiondeep-learning
    View on GitHub↗9,244
See all 30 alternatives to U 2 Net→