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
Back to zijundeng/pytorch-semantic-segmentation

Projects sharing features with Pytorch Semantic Segmentation

30 open-source projects similar to zijundeng/pytorch-semantic-segmentation, ranked by shared indexed features. Tags may describe platforms or build tools rather than the same primary purpose. Check each project’s use case, license, and deployment requirements before treating it as a replacement.

  • facebookresearch/maskrcnn-benchmarkfacebookresearch avatar

    facebookresearch/maskrcnn-benchmark

    9,370View on GitHub↗

    This project is a modular PyTorch framework for training and evaluating object detection and instance segmentation models. It serves as a computer vision research tool and a deep learning inference engine designed to identify object locations, classes, and pixel-level masks within images. The framework implements a two-stage inference pipeline that utilizes region proposal networks and a symmetric mask-head architecture. It provides specialized capabilities for instance segmentation, object bounding box detection, and human pose estimation via anatomical keypoint detection. The system includ

    Python
    View on GitHub↗9,370
  • zhanghang1989/pytorch-style-transferzhanghang1989 avatar

    zhanghang1989/PyTorch-Style-Transfer

    1,007View on GitHub↗

    Neural Style and MSG-Net

    Jupyter Notebook
    View on GitHub↗1,007
  • 1adrianb/face-alignment1adrianb avatar

    1adrianb/face-alignment

    7,518View on GitHub↗

    This is a PyTorch-based computer vision library for detecting 2D and 3D facial landmark coordinates. It functions as a facial landmark detector and reconstruction tool, utilizing deep learning to identify precise geometric points on human faces from image datasets. The library allows for the selection of specific detection backends to balance accuracy and processing speed. It supports the integration of precomputed bounding box files, which enables the system to bypass the initial detection phase and proceed directly to landmark extraction. The toolkit includes capabilities for batch image p

    Python
    View on GitHub↗7,518
  • longcw/yolo2-pytorchlongcw avatar

    longcw/yolo2-pytorch

    1,560View on GitHub↗

    YOLOv2 in PyTorch

    Pythondarknetdetectionpytorch
    View on GitHub↗1,560

AI search

Explore more awesome repositories

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

Find more with AI search
  • ruotianluo/neuraltalk2.pytorchruotianluo avatar

    ruotianluo/neuraltalk2.pytorch

    1,478View on GitHub↗

    I decide to sync up this repo and self-critical.pytorch. (The old master is in old master branch for archive)

    Python
    View on GitHub↗1,478
  • ycszen/torchsegycszen avatar

    ycszen/TorchSeg

    1,409View on GitHub↗

    Fast, modular reference implementation and easy training of Semantic Segmentation algorithms in PyTorch.

    Python
    View on GitHub↗1,409
  • cmu-perceptual-computing-lab/openposeCMU-Perceptual-Computing-Lab avatar

    CMU-Perceptual-Computing-Lab/openpose

    34,145View on GitHub↗

    OpenPose is a real-time pose estimation engine designed to detect and track human body, face, hand, and foot landmarks. It functions as a multi-person motion tracker, identifying the spatial coordinates of multiple individuals simultaneously within video streams or static images. Beyond two-dimensional detection, the software acts as a three-dimensional kinematics processor, reconstructing spatial movement data from single or multiple synchronized camera perspectives. The system distinguishes itself through a bottom-up approach that utilizes part-affinity fields to associate body parts across

    C++caffecomputer-visioncpp
    View on GitHub↗34,145
  • eladhoffer/captiongenE

    eladhoffer/captionGen

    0View on GitHub↗
    View on GitHub↗0
  • hyeonwoonoh/deconvnetHyeonwooNoh avatar

    HyeonwooNoh/DeconvNet

    326View on GitHub↗

    DeconvNet : Learning Deconvolution Network for Semantic Segmentation

    Matlab
    View on GitHub↗326
  • jocicmarko/ultrasound-nerve-segmentationjocicmarko avatar

    jocicmarko/ultrasound-nerve-segmentation

    944View on GitHub↗

    This tutorial shows how to use Keras library to build deep neural network for ultrasound image nerve segmentation. More info on this Kaggle competition can be found on https://www.kaggle.com/c/ultrasound-nerve-segmentation.

    Python
    View on GitHub↗944
  • osmr/imgclsmobosmr avatar

    osmr/imgclsmob

    3,018View on GitHub↗

    This repo is used to research convolutional networks primarily for computer vision tasks. For this purpose, the repo contains (re)implementations of various classification, segmentation, detection, and pose estimation models and scripts for training/evaluating/converting.

    Python
    View on GitHub↗3,018
  • progamergov/neural-style-ptProGamerGov avatar

    ProGamerGov/neural-style-pt

    860View on GitHub↗

    PyTorch implementation of neural style transfer algorithm

    Python
    View on GitHub↗860
  • thnkim/openfacepytorchthnkim avatar

    thnkim/OpenFacePytorch

    189View on GitHub↗

    PyTorch module to use OpenFace's nn4.small2.v1.t7 model

    Python
    View on GitHub↗189
  • wkentaro/pytorch-fcnwkentaro avatar

    wkentaro/pytorch-fcn

    1,777View on GitHub↗

    PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

    Pythoncomputer-visionconvolutional-networksdeep-learning
    View on GitHub↗1,777
  • nvidia/flownet2-pytorchNVIDIA avatar

    NVIDIA/flownet2-pytorch

    3,286View on GitHub↗

    Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

    Python
    View on GitHub↗3,286
  • bodokaiser/piwiseB

    bodokaiser/piwise

    0View on GitHub↗

    Pixel-wise segmentation on the VOC2012dataset dataset using pytorchpytorch.

    View on GitHub↗0
  • divamgupta/image-segmentation-kerasD

    divamgupta/image-segmentation-keras

    0View on GitHub↗

    Implementation of various Deep Image Segmentation models in keras.

    View on GitHub↗0
  • dbolya/yolactdbolya avatar

    dbolya/yolact

    5,231View on GitHub↗

    Yolact is a computer vision framework and real-time instance segmentation model. It utilizes a fully convolutional neural network to detect objects and generate pixel-level masks for images and video feeds. The system employs prototypical mask generation to create global mask prototypes that are linearly combined for instance-specific results. It incorporates deformable convolutional layers and deformable region-of-interest pooling to adapt spatial sampling to the irregular shapes of objects. The framework covers the full model development lifecycle, including training on custom datasets, ac

    Python
    View on GitHub↗5,231
  • facebookresearch/detectron2facebookresearch avatar

    facebookresearch/detectron2

    34,548View on GitHub↗

    Detectron2 is a PyTorch computer vision framework and visual recognition platform designed for training and deploying models for object detection, image segmentation, and visual recognition. It provides a research-oriented environment for training complex vision models with multi-GPU acceleration. The project includes a specialized object detection library for identifying and locating multiple objects via bounding boxes, as well as an image segmentation toolkit for creating pixel-level masks through instance, semantic, and panoptic segmentation. Additionally, it features a human pose estimati

    Python
    View on GitHub↗34,548
  • 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
  • ignacio-rocco/detectorchignacio-rocco avatar

    ignacio-rocco/detectorch

    559View on GitHub↗

    (Disclaimer: this is work in progress and does not feature all the functionalities of detectron. Currently only inference and evaluation are supported -- no training) (News: Now supporting FPN and ResNet-101!)

    Jupyter Notebook
    View on GitHub↗559
  • isht7/pytorch-deeplab-resnetI

    isht7/pytorch-deeplab-resnet

    0View on GitHub↗
    View on GitHub↗0
  • nvidia/semantic-segmentationNVIDIA avatar

    NVIDIA/semantic-segmentation

    1,823View on GitHub↗

    Nvidia Semantic Segmentation monorepo

    Python
    View on GitHub↗1,823
  • open-mmlab/mmdetectionopen-mmlab avatar

    open-mmlab/mmdetection

    32,756View on GitHub↗

    This project is a modular research toolkit designed for developing, training, and evaluating deep learning models for object detection, segmentation, and video instance tracking. It provides a flexible training engine that manages complex neural network execution, including distributed training, custom lifecycle hooks, and weight optimization. The framework is built around a hierarchical configuration system that allows users to define architectures, data pipelines, and training hyperparameters through composable, inheritable files. The project distinguishes itself through its highly modular

    Pythoncascade-rcnnconvnextdetr
    View on GitHub↗32,756
  • amdegroot/ssd.pytorchamdegroot avatar

    amdegroot/ssd.pytorch

    5,224View on GitHub↗

    This is a PyTorch object detection framework that implements the Single Shot MultiBox Detector for identifying and localizing multiple objects within images and video. The project provides a neural network architecture designed for single-shot object detection, which predicts bounding boxes and class labels in one pass. The implementation includes a real-time object detector capable of processing live video streams to track and label objects across sequential frames. It also features a complete computer vision training pipeline for preparing image datasets and training model weights. The fra

    Pythoncomputer-visiondeep-learningimage-recognition
    View on GitHub↗5,224
  • potterhsu/svhnclassifier-pytorchP

    potterhsu/SVHNClassifier-PyTorch

    0View on GitHub↗
    View on GitHub↗0
  • rwightman/pytorch-image-modelsrwightman avatar

    rwightman/pytorch-image-models

    36,893View on GitHub↗

    This project is a library of pretrained computer vision architectures and backbones for image classification and feature extraction. It serves as a comprehensive model zoo and collection of standardized image encoders, including ResNet, Vision Transformers, and EfficientNet, for use in visual analysis and as backbones for object detection and image segmentation. The library provides a framework for distributed training and evaluation of image models using advanced data augmentation and optimization scripts. It includes a dedicated toolset for converting trained PyTorch vision models into the

    Python
    View on GitHub↗36,893
  • tensorflow/modelstensorflow avatar

    tensorflow/models

    77,663View on GitHub↗

    This repository serves as a centralized collection of state-of-the-art deep learning architectures and reference implementations designed for research and application development. It provides a comprehensive toolkit for computer vision and natural language processing, offering pre-built models and training pipelines for tasks ranging from image classification and object detection to complex sequence modeling. The project distinguishes itself by providing a flexible execution harness that manages the entire training lifecycle, including data ingestion and backpropagation. It supports scalable

    Python
    View on GitHub↗77,663
  • thstkdgus35/edsr-pytorchthstkdgus35 avatar

    thstkdgus35/EDSR-PyTorch

    2,627View on GitHub↗

    About PyTorch 1.2.0 Now the master branch supports PyTorch 1.2.0 by default. Due to the serious version problem (especially torch.utils.data.dataloader), MDSR functions are temporarily disabled. If you have to train/evaluate the MDSR model, please use legacy branches.

    Python
    View on GitHub↗2,627
  • cs230-stanford/cs230-code-examplescs230-stanford avatar

    cs230-stanford/cs230-code-examples

    4,218View on GitHub↗

    This repository provides structured code examples and project templates designed for classroom instruction in machine learning and neural networks. It offers reference implementations of deep learning models for both computer vision and natural language processing tasks, built using PyTorch as the core framework. The codebase is organized as a modular project template with separate directories for data handling, model definitions, and training scripts, promoting reusability and clarity. It includes predefined pipelines for image classification and text processing, along with a command-line in

    Pythoncomputer-visionnatural-language-processingpytorch
    View on GitHub↗4,218