Awesome Open Source

Programming Languages

Search results for pytorch segmentation

576 search results found

Deep Learning For Image Processing ⭐ 18,759

deep learning for image processing including classification and object-detection etc.

Segmentation_models.pytorch ⭐ 8,226

Segmentation models with pretrained backbones. PyTorch.

Pytorch Unet ⭐ 7,861

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

Mmsegmentation ⭐ 6,974

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Pytorch Cnn Visualizations ⭐ 6,883

Pytorch implementation of convolutional neural network visualization techniques

Yolact ⭐ 4,793

A simple, fully convolutional model for real-time instance segmentation.

Nnunet ⭐ 4,608

Catalyst ⭐ 3,151

Accelerated deep learning R&D

Resnest ⭐ 3,070

ResNeSt: Split-Attention Networks

Detectron.pytorch ⭐ 2,695

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.

Torch Points3d ⭐ 2,550

Pytorch framework for doing deep learning on point clouds.

Pointnet_pointnet2_pytorch ⭐ 2,491

PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.

Imgclsmob ⭐ 2,399

Sandbox for training deep learning networks

Awesome Semantic Segmentation Pytorch ⭐ 2,399

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)

Deep Learning For Tracking And Detection ⭐ 2,270

Collection of papers, datasets, code and other resources for object tracking and detection using deep learning

Image_segmentation ⭐ 2,186

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

Pointnet.pytorch ⭐ 1,759

pytorch implementation for "PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation" https://arxiv.org/abs/1612.00593

3dunetcnn ⭐ 1,750

Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation

Pytorch Networks ⭐ 1,740

Pytorch implementation of cnn network

Pytorch 3dunet ⭐ 1,721

3D U-Net model for volumetric semantic segmentation written in pytorch

Ailia Models ⭐ 1,708

The collection of pre-trained, state-of-the-art AI models for ailia SDK

Detrex ⭐ 1,644

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.

Danet ⭐ 1,556

Dual Attention Network for Scene Segmentation (CVPR2019)

Ml Cvnets ⭐ 1,543

CVNets: A library for training computer vision networks

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model

Tutorials ⭐ 1,489

MONAI Tutorials

Pytorch Toolbelt ⭐ 1,458

PyTorch extensions for fast R&D prototyping and Kaggle farming

Dlcv_for_beginners ⭐ 1,446

《深度学习与计算机视觉》配套代码

Unet Segmentation Pytorch Nest Of Unets ⭐ 1,236

Implementation of different kinds of Unet Models for Image Segmentation - Unet , RCNN-Unet, Attention Unet, RCNN-Attention Unet, Nested Unet

Mmrazor ⭐ 1,231

OpenMMLab Model Compression Toolbox and Benchmark.

Meshcnn ⭐ 1,211

Convolutional Neural Network for 3D meshes in PyTorch

Dgcnn ⭐ 1,179

Vocal Remover ⭐ 1,148

Vocal Remover using Deep Neural Networks

Yolact_edge ⭐ 1,138

The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.

Torchseg ⭐ 1,106

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

Ccnet ⭐ 1,061

CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).

Unet Family ⭐ 1,029

Paper and implementation of UNet-related model.

Deeplabv3plus Pytorch ⭐ 1,010

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

[CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers

Discogan Pytorch ⭐ 928

PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"

Dilated Residual Networks

alfred-py: A deep learning utility library for **human**, more detail about the usage of lib to: https://zhuanlan.zhihu.com/p/341446046

Medicalzoopytorch ⭐ 896

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

Image Test Time Augmentation with PyTorch!

Semantic Segmentation in Pytorch

Pixellib ⭐ 828

Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

Lightnet ⭐ 715

LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)

Pytorch Auto Drive ⭐ 713

PytorchAutoDrive: Segmentation models (ERFNet, ENet, DeepLab, FCN...) and Lane detection models (SCNN, RESA, LSTR, LaneATT, BézierLaneNet...) based on PyTorch with fast training, visualization, benchmarking & deployment help

Superpoint_graph ⭐ 691

Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs

Kpconv Pytorch ⭐ 681

Kernel Point Convolution implemented in PyTorch

Mobile Semantic Segmentation ⭐ 677

Real-Time Semantic Segmentation in Mobile device

FSGAN - Official PyTorch Implementation

Ritm_interactive_segmentation ⭐ 629

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

Cascadepsp ⭐ 622

[CVPR 2020] CascadePSP: Toward Class-Agnostic and Very High-Resolution Segmentation via Global and Local Refinement

Pytorch Segmentation Toolbox ⭐ 616

PyTorch Implementations for DeeplabV3 and PSPNet

All About The Gan ⭐ 602

All About the GANs(Generative Adversarial Networks) - Summarized lists for GAN

EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

Adaptsegnet ⭐ 586

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)

Fbrs_interactive_segmentation ⭐ 584

[CVPR2020] f-BRS: Rethinking Backpropagating Refinement for Interactive Segmentation https://arxiv.org/abs/2001.10331

Brain Segmentation Pytorch ⭐ 554

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI

Pytorch Medical Segmentation ⭐ 536

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.

常用的语义分割架构结构综述以及代码复现

Polarmask ⭐ 522

Code for 'PolarMask: Single Shot Instance Segmentation with Polar Representation'

Ternausnetv2 ⭐ 500

TernausNetV2: Fully Convolutional Network for Instance Segmentation

[NeurIPS 2021] Rethinking Space-Time Networks with Improved Memory Coverage for Efficient Video Object Segmentation

Monailabel ⭐ 496

MONAI Label is an intelligent open source image labeling and learning tool.

Medical Transformer ⭐ 484

Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021

Dgcnn.pytorch ⭐ 453

A PyTorch implementation of Dynamic Graph CNN for Learning on Point Clouds (DGCNN)

Osvos Pytorch ⭐ 434

PyTorch implementation of One-Shot Video Object Segmentation (OSVOS)

[CVPR 2021] Modular Interactive Video Object Segmentation: Interaction-to-Mask, Propagation and Difference-Aware Fusion. Semi-supervised VOS as well!

LEDNet: A Lightweight Encoder-Decoder Network for Real-time Semantic Segmentation

Vnet.pytorch ⭐ 421

A PyTorch implementation for V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation

Lanedetection_end2end ⭐ 405

End-to-end Lane Detection for Self-Driving Cars (ICCV 2019 Workshop)

Point Transformer Pytorch ⭐ 402

Implementation of the Point Transformer layer, in Pytorch

Pytorch Hair Segmentation ⭐ 399

pytorch-hair-segmentation

Vedaseg ⭐ 383

A semantic segmentation toolbox based on PyTorch

UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation.

Pspnet Pytorch ⭐ 377

PyTorch implementation of PSPNet segmentation network

Multi Task Learning Pytorch ⭐ 377

PyTorch implementation of multi-task learning architectures, incl. MTI-Net (ECCV2020).

Espnetv2 ⭐ 375

A light-weight, power efficient, and general purpose convolutional neural network

Real Time Network ⭐ 374

real-time network architecture for mobile devices and semantic segmentation

Fcn Pytorch ⭐ 366

🚘 Easiest Fully Convolutional Networks

High-level batteries-included neural network training library for Pytorch

Pytorch Unet ⭐ 357

Simple PyTorch implementations of U-Net/FullyConvNet (FCN) for image segmentation

Pytorch Cpp ⭐ 353

Pytorch C++ Library

Pixel-wise segmentation on VOC2012 dataset using pytorch.

Autodeeplab ⭐ 342

Pytorch Implementation the paper Auto-DeepLab Hierarchical Neural Architecture Search for Semantic Image Segmentation

Edgenets ⭐ 340

This repository contains the source code of our work on designing efficient CNNs for computer vision

Panoptic Deeplab ⭐ 333

This is Pytorch re-implementation of our CVPR 2020 paper "Panoptic-DeepLab: A Simple, Strong, and Fast Baseline for Bottom-Up Panoptic Segmentation" (https://arxiv.org/abs/1911.10194)

Pytorch Goodies ⭐ 332

PyTorch Boilerplate For Research

Efficient Segmentation Networks ⭐ 323

Lightweight models for real-time semantic segmentationon PyTorch (include SQNet, LinkNet, SegNet, UNet, ENet, ERFNet, EDANet, ESPNet, ESPNetv2, LEDNet, ESNet, FSSNet, CGNet, DABNet, Fast-SCNN, ContextNet, FPENet, etc.)

Advsemiseg ⭐ 307

Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018

Dc Unet ⭐ 298

We proposed a novel U-Net-based model -- DC-UNet to do medical image segmentation.

Uctransnet ⭐ 293

Implementation of our AAAI'22 work: 'UCTransNet: Rethinking the Skip Connections in U-Net from a Channel-wise Perspective with Transformer'.

Torchsat ⭐ 285

🔥TorchSat 🌏 is an open-source deep learning framework for satellite imagery analysis based on PyTorch.

Pytorch Segmentation ⭐ 284

PyTorch implementation for semantic segmentation (DeepLabV3+, UNet, etc.)

Cfpnet Medicine ⭐ 271

CFPNet-M: A Light-Weight Encoder-Decoder Based Network for Multimodal Biomedical Image Real-Time Segmentation

[NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "Fast Vision Transformers with HiLo Attention"

Bisenet ⭐ 266

BiSeNet based on pytorch

Related Searches

Python Pytorch (15,118)

Deep Learning Pytorch (6,344)

Jupyter Notebook Pytorch (4,892)

Python Segmentation (4,252)

Machine Learning Pytorch (2,934)

Dataset Pytorch (1,847)

Pytorch Convolutional Neural Networks (1,567)

Pytorch Natural Language Processing (1,408)

Pytorch Neural Network (1,391)

Jupyter Notebook Segmentation (1,309)

1-100 of 576 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2025 Awesome Open Source. All rights reserved.