GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → tsunghan-mama → RandLA-Net-pytorch

tsunghan-mama / RandLA-Net-pytorch

Licence: MIT license
🍀 Pytorch Implementation of RandLA-Net (https://arxiv.org/abs/1911.11236)

Programming Languages

C++
36643 projects - #6 most used programming language
python
139335 projects - #7 most used programming language
cython
566 projects

Labels

computer-visionsemantic-segmentation3d-visionpytorch-implementationsemantickitti

Projects that are alternatives of or similar to RandLA-Net-pytorch

3D-UNet-PyTorch-Implementation
The implementation of 3D-UNet using PyTorch
Stars: ✭ 78 (+13.04%)
Mutual labels:  semantic-segmentation, pytorch-implementation
label-fusion
Volumetric Fusion of Multiple Semantic Labels and Masks
Stars: ✭ 18 (-73.91%)
Mutual labels:  semantic-segmentation, 3d-vision
lowshot-shapebias
Learning low-shot object classification with explicit shape bias learned from point clouds
Stars: ✭ 37 (-46.38%)
Mutual labels:  3d-vision, pytorch-implementation
VT-UNet
[MICCAI2022] This is an official PyTorch implementation for A Robust Volumetric Transformer for Accurate 3D Tumor Segmentation
Stars: ✭ 151 (+118.84%)
Mutual labels:  semantic-segmentation, pytorch-implementation
DocuNet
Code and dataset for the IJCAI 2021 paper "Document-level Relation Extraction as Semantic Segmentation".
Stars: ✭ 84 (+21.74%)
Mutual labels:  semantic-segmentation, pytorch-implementation
ResUNetPlusPlus-with-CRF-and-TTA
ResUNet++, CRF, and TTA for segmentation of medical images (IEEE JBIHI)
Stars: ✭ 98 (+42.03%)
Mutual labels:  semantic-segmentation, pytorch-implementation
MolDQN-pytorch
A PyTorch Implementation of "Optimization of Molecules via Deep Reinforcement Learning".
Stars: ✭ 58 (-15.94%)
Mutual labels:  pytorch-implementation
Fashion-Clothing-Parsing
FCN, U-Net models implementation in TensorFlow for fashion clothing parsing
Stars: ✭ 29 (-57.97%)
Mutual labels:  semantic-segmentation
ssgan
Semi Supervised Semantic Segmentation Using Generative Adversarial Network ; Pytorch
Stars: ✭ 25 (-63.77%)
Mutual labels:  semantic-segmentation
RGBD-semantic-segmentation
A paper list of RGBD semantic segmentation (processing)
Stars: ✭ 264 (+282.61%)
Mutual labels:  semantic-segmentation
smartImgProcess
手工实现的智能图片处理系统 包含基础的图片处理功能 各类滤波 seam carving算法 以及结合精细语义分割信息 实现智能去除目标的功能
Stars: ✭ 22 (-68.12%)
Mutual labels:  semantic-segmentation
CSSR
Crack Segmentation for Low-Resolution Images using Joint Learning with Super-Resolution (CSSR) was accepted to international conference on MVA2021 (oral), and selected for the Best Practical Paper Award.
Stars: ✭ 50 (-27.54%)
Mutual labels:  semantic-segmentation
NGCF-PyTorch
PyTorch Implementation for Neural Graph Collaborative Filtering
Stars: ✭ 200 (+189.86%)
Mutual labels:  pytorch-implementation
generative pose
Code for our ICCV 19 paper : Monocular 3D Human Pose Estimation by Generation and Ordinal Ranking
Stars: ✭ 63 (-8.7%)
Mutual labels:  3d-vision
ScasNet
Semantic Labeling in VHR Images via A Self-Cascaded CNN (ISPRS JPRS, IF=6.942)
Stars: ✭ 24 (-65.22%)
Mutual labels:  semantic-segmentation
MinkLocMultimodal
MinkLoc++: Lidar and Monocular Image Fusion for Place Recognition
Stars: ✭ 65 (-5.8%)
Mutual labels:  3d-vision
tf-semantic-segmentation-FCN-VGG16
Semantic segmentation for classifying road. "Fully Convolutional Networks for Semantic Segmentation (2015)" implemented using TF
Stars: ✭ 30 (-56.52%)
Mutual labels:  semantic-segmentation
Representation-Learning-for-Information-Extraction
Pytorch implementation of Paper by Google Research - Representation Learning for Information Extraction from Form-like Documents.
Stars: ✭ 82 (+18.84%)
Mutual labels:  pytorch-implementation
onn
Online Deep Learning: Learning Deep Neural Networks on the Fly / Non-linear Contextual Bandit Algorithm (ONN_THS)
Stars: ✭ 139 (+101.45%)
Mutual labels:  pytorch-implementation
pysemseg
Semantic Segmentation Models in Pytorch
Stars: ✭ 31 (-55.07%)
Mutual labels:  semantic-segmentation
View All Similar Projects ➔

RandLA-Net-pytorch

This repository contains the implementation of RandLA-Net (CVPR 2020 Oral) in PyTorch.

  • We only support SemanticKITTI dataset now. (Welcome everyone to develop together and raise PR)
  • Our model is almost as good as the original implementation. (Validation set : Our 52.9% mIoU vs original 53.1%)
  • We place our pretrain-model in pretrain_model/checkpoint.tar directory.

Performance

Results on Validation Set (seq 08)

  • Compare with original implementation
Model mIoU
Original Tensorflow 0.531
Our Pytorch Implementation 0.529
  • Per class mIoU
mIoU car bicycle motorcycle truck other-vehicle person bicyclist motorcyclist road parking sidewalk other-ground building fence vegetation trunk terrain pole traffic-sign
52.9 0.919 0.122 0.290 0.660 0.444 0.515 0.676 0.000 0.912 0.421 0.759 0.001 0.878 0.354 0.844 0.595 0.741 0.517 0.414

A. Environment Setup

  1. Click this webpage and use conda to install pytorch>=1.4 (Be aware of the cuda version when installation)

  2. Install python packages

pip install -r requirements.txt
  1. Compile C++ Wrappers
bash compile_op.sh

B. Prepare Data

Download the Semantic KITTI dataset, and preprocess the data:

python data_prepare_semantickitti.py

Note:

  • Please change the dataset path in the data_prepare_semantickitti.py with your own path.
  • Data preprocessing code will convert the label to 0-19 index

C. Training & Testing

  1. Training
python3 train_SemanticKITTI.py <args>
  1. Testing
python3 test_SemanticKITTI.py <args>

Note: if the flag --index_to_label is set, output predictions will be ".label" files (label figure) which can be visualized; Otherwise, they will be ".npy" (0-19 index) files which is used to evaluated afterward.

D. Visualization & Evaluation

  1. Visualization
python3 visualize_SemanticKITTI.py <args>
  1. Evaluation
  • Example Evaluation code
python3 evaluate_SemanticKITTI.py --dataset /tmp2/tsunghan/PCL_Seg_data/sequences_0.06/ \
    --predictions runs/supervised/predictions/ --sequences 8

Acknowledgement

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