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

milesial/Pytorch-UNet

0
View on GitHub↗

Pytorch UNet

Pytorch-UNet is a deep learning implementation designed for semantic image segmentation. It provides a framework for training convolutional neural networks to perform pixel-wise classification, transforming input images into detailed prediction masks.

The project utilizes a symmetric encoder-decoder architecture that employs skip-connection feature fusion to recover fine-grained boundary details. It includes support for mixed-precision training to reduce memory usage and accelerate processing speeds.

The framework covers the end-to-end segmentation pipeline, from model training using custom datasets to the generation of masks through a command-line tool. It also includes real-time monitoring for tracking loss curves and validation metrics, with specific applications for medical image analysis and image masking competitions.

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

Features

  • PyTorch Semantic Segmentation Libraries - Implements a specialized PyTorch framework for semantic image segmentation using an encoder-decoder architecture.
  • Computer Vision Models - Implements a convolutional neural network designed for image segmentation and object boundary detection.
  • Image Segmenters - Provides a set of tools utilizing convolutional neural networks for pixel-level object isolation.
  • Symmetric Encoder-Decoders - Features a symmetric encoder-decoder structure that contracts resolution for context and expands it for precise mask reconstruction.
  • U-Net Architectures - Implements the U-Net architecture in PyTorch for automated semantic image segmentation.
  • Skip-Connection Architectures - Utilizes skip-connections to pass high-resolution spatial information from early layers to deep layers for boundary recovery.
  • Semantic Masking Architectures - Implements an architecture that assigns semantic class labels to every individual pixel for detailed image partitioning.
  • Pixel-Level Classifiers - Provides a neural network architecture that treats each individual pixel as a classification target.
  • Image Masking Frameworks - Provides a framework for performing pixel-level classification to isolate objects and create precise prediction masks.
  • Semantic Segmentation Training - Provides a training process for semantic segmentation models using custom datasets and mixed precision.
  • Deep Learning Training Toolsets - Provides a toolset for training and optimizing deep neural networks with mixed precision and real-time metric logging.
  • Mixed Precision Training - Employs a combination of 16-bit and 32-bit floating point numbers to accelerate training and reduce GPU memory usage.
  • Training Progress Monitoring - Includes a real-time logging system for tracking loss curves and validation metrics during model training.
  • Medical Image Segmentations - Supports the isolation of anatomical structures in high-resolution medical scans through deep learning segmentation.
  • Mixed Precision Training Utilities - Provides utilities for managing precision and gradient scaling to improve memory and computational efficiency during training.
  • Pixel-Wise Loss Functions - Implements loss functions that calculate the difference between predicted segmentation masks and ground truth labels.
  • Prediction Mask Generators - Ships a command-line tool to process images through a trained model and save the resulting semantic masks.
  • Neural Network Architectures - PyTorch implementation of the U-Net architecture for segmentation.
  • Image segmentation - Listed in the “Image segmentation” section of the Ailia Models awesome list.
11,503 stars·2,752 forks·Python·GPL-3.0·20 views

Star history

Star history chart for milesial/pytorch-unetStar history chart for milesial/pytorch-unet

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.

Frequently asked questions

What does milesial/pytorch-unet do?

Pytorch-UNet is a deep learning implementation designed for semantic image segmentation. It provides a framework for training convolutional neural networks to perform pixel-wise classification, transforming input images into detailed prediction masks.

What are the main features of milesial/pytorch-unet?

The main features of milesial/pytorch-unet are: PyTorch Semantic Segmentation Libraries, Computer Vision Models, Image Segmenters, Symmetric Encoder-Decoders, U-Net Architectures, Skip-Connection Architectures, Semantic Masking Architectures, Pixel-Level Classifiers.

Which projects share features with milesial/pytorch-unet?

Projects with overlapping indexed features include: zhixuhao/unet — This project is a PyTorch implementation of a U-Net convolutional neural network designed for pixel-level image… leoxiaobin/deep-high-resolution-net.pytorch — This project is a PyTorch implementation of a research architecture designed for high-resolution representation… fastai/course-v3 — This repository is a comprehensive educational program and deep learning framework designed to teach practical deep… qubvel-org/segmentation_models.pytorch — This is a PyTorch semantic segmentation library designed for building image masking frameworks. It provides a… casia-lmc-lab/fastsam — FastSAM is an image segmentation framework that uses convolutional neural networks to isolate visual elements and… leejunhyun/image_segmentation — This project is a biomedical image segmentation framework and PyTorch computer vision library. It provides a deep…

Projects sharing features with Pytorch UNet

These projects share indexed features with Pytorch UNet. Shared tags can include platform or build tooling; verify the primary use case before treating a result as a replacement.
  • zhixuhao/unetzhixuhao avatar

    zhixuhao/unet

    4,928View on GitHub↗

    This project is a PyTorch implementation of a U-Net convolutional neural network designed for pixel-level image segmentation. It functions as a biomedical image processor that generates precise masks to isolate anatomical structures within medical imagery. The architecture utilizes a symmetric encoder-decoder structure to capture context and enable precise localization. It employs skip-connection feature fusion to combine high-resolution features from the contracting path with upsampled outputs, recovering spatial detail. The system covers deep learning model training using binary cross-entr

    Jupyter Notebookkerassegmentationunet
    View on GitHub↗4,928
  • leoxiaobin/deep-high-resolution-net.pytorchleoxiaobin avatar

    leoxiaobin/deep-high-resolution-net.pytorch

    4,479View on GitHub↗

    This project is a PyTorch implementation of a research architecture designed for high-resolution representation learning. It serves as a computer vision framework focused on precise keypoint detection, human pose estimation, and semantic image segmentation. The implementation provides specialized tools for identifying anatomical landmarks on the human body and predicting facial keypoint coordinates to analyze orientation and alignment. It utilizes a system of multi-resolution parallel streams and repeated multi-scale fusion to maintain high-resolution representations throughout the network.

    Cuda
    View on GitHub↗4,479
  • fastai/course-v3fastai avatar

    fastai/course-v3

    4,914View on GitHub↗

    This repository is a comprehensive educational program and deep learning framework designed to teach practical deep learning using PyTorch through notebooks and code examples. It serves as a high-level library for building, training, and deploying neural networks, acting as a model training orchestrator that coordinates PyTorch models, optimizers, and loss functions. The project provides specialized toolkits for computer vision, natural language processing, and tabular data preprocessing. It distinguishes itself through advanced training controls such as discriminative learning rates, a two-w

    Jupyter Notebookdata-sciencedeep-learningfastai
    View on GitHub↗4,914
  • qubvel-org/segmentation_models.pytorchqubvel-org avatar

    qubvel-org/segmentation_models.pytorch

    11,622View on GitHub↗

    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 al

    Pythoncomputer-visiondeeplab-v3-plusdeeplabv3
    View on GitHub↗11,622
Compare all 30 related projects→