All Projects → shahabty → PSPNet-Pytorch

shahabty / PSPNet-Pytorch

Licence: other
Implemetation of Pyramid Scene Parsing Network in Pytorch

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to PSPNet-Pytorch

Segmentation models.pytorch
Segmentation models with pretrained backbones. PyTorch.
Stars: ✭ 4,584 (+17530.77%)
Mutual labels:  semantic-segmentation, pspnet
Vpgnet
VPGNet: Vanishing Point Guided Network for Lane and Road Marking Detection and Recognition (ICCV 2017)
Stars: ✭ 382 (+1369.23%)
Mutual labels:  caffe, semantic-segmentation
Mmsegmentation
OpenMMLab Semantic Segmentation Toolbox and Benchmark.
Stars: ✭ 2,875 (+10957.69%)
Mutual labels:  semantic-segmentation, pspnet
adversarial-attacks
Code for our CVPR 2018 paper, "On the Robustness of Semantic Segmentation Models to Adversarial Attacks"
Stars: ✭ 90 (+246.15%)
Mutual labels:  caffe, semantic-segmentation
Scale-Adaptive-Network
Semantic Image Segmentation by Scale-Adaptive Networks (TIP 2019)
Stars: ✭ 42 (+61.54%)
Mutual labels:  caffe, semantic-segmentation
TensorFlow-Advanced-Segmentation-Models
A Python Library for High-Level Semantic Segmentation Models based on TensorFlow and Keras with pretrained backbones.
Stars: ✭ 64 (+146.15%)
Mutual labels:  semantic-segmentation, pspnet
Cascaded Fcn
Source code for the MICCAI 2016 Paper "Automatic Liver and Lesion Segmentation in CT Using Cascaded Fully Convolutional NeuralNetworks and 3D Conditional Random Fields"
Stars: ✭ 296 (+1038.46%)
Mutual labels:  caffe, semantic-segmentation
semantic-segmentation-tensorflow
Semantic segmentation task for ADE20k & cityscapse dataset, based on several models.
Stars: ✭ 84 (+223.08%)
Mutual labels:  semantic-segmentation, pspnet
Pixelnet
The repository contains source code and models to use PixelNet architecture used for various pixel-level tasks. More details can be accessed at <http://www.cs.cmu.edu/~aayushb/pixelNet/>.
Stars: ✭ 194 (+646.15%)
Mutual labels:  caffe, semantic-segmentation
Jacinto Ai Devkit
Training & Quantization of embedded friendly Deep Learning / Machine Learning / Computer Vision models
Stars: ✭ 49 (+88.46%)
Mutual labels:  caffe, semantic-segmentation
Enet
ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentation
Stars: ✭ 530 (+1938.46%)
Mutual labels:  caffe, semantic-segmentation
image-segmentation
Mask R-CNN, FPN, LinkNet, PSPNet and UNet with multiple backbone architectures support readily available
Stars: ✭ 62 (+138.46%)
Mutual labels:  semantic-segmentation, pspnet
Dsrg
Weakly-Supervised Semantic Segmentation Network with Deep Seeded Region Growing (CVPR 2018).
Stars: ✭ 206 (+692.31%)
Mutual labels:  caffe, semantic-segmentation
etos-deepcut
Deep Extreme Cut http://www.vision.ee.ethz.ch/~cvlsegmentation/dextr . a tool to do automatically object segmentation from extreme points.
Stars: ✭ 24 (-7.69%)
Mutual labels:  semantic-segmentation, pspnet
SEC-tensorflow
a tensorflow version for SEC approach in the paper "seed, expand and constrain: three principles for weakly-supervised image segmentation".
Stars: ✭ 35 (+34.62%)
Mutual labels:  semantic-segmentation
vgg16 batchnorm
VGG16 architecture with BatchNorm
Stars: ✭ 14 (-46.15%)
Mutual labels:  caffe
pysemseg
Semantic Segmentation Models in Pytorch
Stars: ✭ 31 (+19.23%)
Mutual labels:  semantic-segmentation
caffe weight converter
Caffe-to-Keras weight converter. Can also export weights as Numpy arrays for further processing.
Stars: ✭ 68 (+161.54%)
Mutual labels:  caffe
caffe-simnets
The SimNets Architecture's Implementation in Caffe
Stars: ✭ 13 (-50%)
Mutual labels:  caffe
RandLA-Net-pytorch
🍀 Pytorch Implementation of RandLA-Net (https://arxiv.org/abs/1911.11236)
Stars: ✭ 69 (+165.38%)
Mutual labels:  semantic-segmentation

PSPNet-Pytorch

This Repo contains an implemetation of "Pyramid Scene Parsing Network" in Pytorch. Pretrained weights are converted from Official Caffe Repo. The performance is: [acc 0.95196], [acc_cls 0.83369], [mean_iu 0.75547], [fwavacc 0.91172] on Validation set of CityScapes Dataset.

Installation

  1. Install Anaconda3 from Here
  2. Create a Conda Environment:
username@PC:~$ conda env create -f environment.yml
  1. Download Cityscapes Dataset from Here
  2. Download Caffe Pretrained from Here and put it in Caffe-PSPNet folder
  3. Run the code:
username@PC:~$ python main.py

Qualitative

Note

Weight conversion from Caffe to Pytorch is modified from pytorch-semseg.
Preprocessing and loss function is modified from pytorch-semantic-segmentation.
Since the differences are significant, I decided not to add a branch to any of the the above project.

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].