All Projects → charlesCXK → Pytorch_semantic_segmentation

charlesCXK / Pytorch_semantic_segmentation

Licence: mit
Implement some models of RGB/RGBD semantic segmentation in PyTorch, easy to run. Such as FCN, RefineNet, PSPNet, RDFNet, 3DGNN, PointNet, DeepLab V3, DeepLab V3 plus, DenseASPP, FastFCN

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Pytorch semantic segmentation

Vision4j Collection
Collection of computer vision models, ready to be included in a JVM project
Stars: ✭ 132 (-3.65%)
Mutual labels:  classification, semantic-segmentation
Labelme
Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).
Stars: ✭ 7,742 (+5551.09%)
Mutual labels:  classification, semantic-segmentation
Torchsat
🔥TorchSat 🌏 is an open-source deep learning framework for satellite imagery analysis based on PyTorch.
Stars: ✭ 261 (+90.51%)
Mutual labels:  classification, semantic-segmentation
Nncf
PyTorch*-based Neural Network Compression Framework for enhanced OpenVINO™ inference
Stars: ✭ 218 (+59.12%)
Mutual labels:  classification, semantic-segmentation
Frostnet
FrostNet: Towards Quantization-Aware Network Architecture Search
Stars: ✭ 85 (-37.96%)
Mutual labels:  classification, semantic-segmentation
flexinfer
A flexible Python front-end inference SDK based on TensorRT
Stars: ✭ 83 (-39.42%)
Mutual labels:  classification, semantic-segmentation
Pywick
High-level batteries-included neural network training library for Pytorch
Stars: ✭ 320 (+133.58%)
Mutual labels:  classification, semantic-segmentation
Geospatial Machine Learning
A curated list of resources focused on Machine Learning in Geospatial Data Science.
Stars: ✭ 289 (+110.95%)
Mutual labels:  classification, semantic-segmentation
Raster Vision
An open source framework for deep learning on satellite and aerial imagery.
Stars: ✭ 1,248 (+810.95%)
Mutual labels:  classification, semantic-segmentation
Pointcnn
PointCNN: Convolution On X-Transformed Points (NeurIPS 2018)
Stars: ✭ 1,120 (+717.52%)
Mutual labels:  classification, pointcloud
Imgclsmob
Sandbox for training deep learning networks
Stars: ✭ 2,405 (+1655.47%)
Mutual labels:  classification, semantic-segmentation
Segmentation
Tensorflow implementation : U-net and FCN with global convolution
Stars: ✭ 101 (-26.28%)
Mutual labels:  classification, semantic-segmentation
Dgm
Direct Graphical Models (DGM) C++ library, a cross-platform Conditional Random Fields library, which is optimized for parallel computing and includes modules for feature extraction, classification and visualization.
Stars: ✭ 157 (+14.6%)
Mutual labels:  classification, semantic-segmentation
InstantDL
InstantDL: An easy and convenient deep learning pipeline for image segmentation and classification
Stars: ✭ 33 (-75.91%)
Mutual labels:  classification, semantic-segmentation
Semantic Segmentation Editor
Web labeling tool for bitmap images and point clouds
Stars: ✭ 894 (+552.55%)
Mutual labels:  pointcloud, semantic-segmentation
Universal Data Tool
Collaborate & label any type of data, images, text, or documents, in an easy web interface or desktop app.
Stars: ✭ 1,356 (+889.78%)
Mutual labels:  classification, semantic-segmentation
Autoannotationtool
A label tool aim to reduce semantic segmentation label time, rectangle and polygon annotation is supported
Stars: ✭ 113 (-17.52%)
Mutual labels:  classification, semantic-segmentation
Dise Domain Invariant Structure Extraction
Pytorch Implementation -- All about Structure: Adapting Structural Information across Domains for Boosting Semantic Segmentation, CVPR 2019
Stars: ✭ 129 (-5.84%)
Mutual labels:  semantic-segmentation
Lsoftmax Pytorch
The Pytorch Implementation of L-Softmax
Stars: ✭ 133 (-2.92%)
Mutual labels:  classification
Regnet
Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"
Stars: ✭ 129 (-5.84%)
Mutual labels:  classification

PyTorch_Semantic_Segmentation

Implement some models of semantic segmentation in PyTorch, easy to run.

Introduction

  1. This repo includes some networks for Semantic Segmentation implemented in pytorch 1.0.0 and python3. See each directory for more information.
  2. I only provide architecture of network here. Dataset and train/test files aren't available here, for I think it can be added according to individual needs.
  3. The code file containing the network structure can be run directly with the set simulation data.
  4. ResNet101 used in this repo is the one which PSPNet used. The difference between this resnet and the original resnet is that the first 7*7 conv layer in the old version is replaced by three small-kernel convs. Pretrained model can be downloaded from here.

Has finished

FCN8s

RefineNet (CVPR 2017)

PSPNet (CVPR 2017)

PointNet (CVPR 2017)

RDFNet (ICCV 2017)

3DGNN (ICCV 2017)

DeepLab V3

DeepLab V3+ (ECCV 2018)

DenseASPP (CVPR 2018)

FastFCN (Arxiv 2019)

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].