All Projects → yanx27 → 3dgnn_pytorch

yanx27 / 3dgnn_pytorch

Licence: mit
3D Graph Neural Networks for RGBD Semantic Segmentation

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to 3dgnn pytorch

Cilantro
A lean C++ library for working with point cloud data
Stars: ✭ 577 (+208.56%)
Mutual labels:  point-cloud, segmentation, rgbd
Torch Points3d
Pytorch framework for doing deep learning on point clouds.
Stars: ✭ 1,135 (+506.95%)
Mutual labels:  point-cloud, segmentation
Pointcnn
PointCNN: Convolution On X-Transformed Points (NeurIPS 2018)
Stars: ✭ 1,120 (+498.93%)
Mutual labels:  point-cloud, segmentation
3d Pointcloud
Papers and Datasets about Point Cloud.
Stars: ✭ 179 (-4.28%)
Mutual labels:  point-cloud, segmentation
3dmatch Toolbox
3DMatch - a 3D ConvNet-based local geometric descriptor for aligning 3D meshes and point clouds.
Stars: ✭ 571 (+205.35%)
Mutual labels:  point-cloud, rgbd
Depth clustering
🚕 Fast and robust clustering of point clouds generated with a Velodyne sensor.
Stars: ✭ 657 (+251.34%)
Mutual labels:  point-cloud, segmentation
Rgbd semantic segmentation pytorch
PyTorch Implementation of some RGBD Semantic Segmentation models.
Stars: ✭ 84 (-55.08%)
Mutual labels:  segmentation, rgbd
Co Fusion
Co-Fusion: Real-time Segmentation, Tracking and Fusion of Multiple Objects
Stars: ✭ 400 (+113.9%)
Mutual labels:  segmentation, rgbd
Gacnet
Pytorch implementation of 'Graph Attention Convolution for Point Cloud Segmentation'
Stars: ✭ 103 (-44.92%)
Mutual labels:  point-cloud, segmentation
Pointnet Keras
Keras implementation for Pointnet
Stars: ✭ 110 (-41.18%)
Mutual labels:  point-cloud, segmentation
Grid Gcn
Grid-GCN for Fast and Scalable Point Cloud Learning
Stars: ✭ 143 (-23.53%)
Mutual labels:  point-cloud, segmentation
Superpoint graph
Large-scale Point Cloud Semantic Segmentation with Superpoint Graphs
Stars: ✭ 533 (+185.03%)
Mutual labels:  point-cloud, segmentation
Pointasnl
PointASNL: Robust Point Clouds Processing using Nonlocal Neural Networks with Adaptive Sampling (CVPR 2020)
Stars: ✭ 159 (-14.97%)
Mutual labels:  point-cloud, segmentation
Peac
Fast Plane Extraction Using Agglomerative Hierarchical Clustering (AHC)
Stars: ✭ 51 (-72.73%)
Mutual labels:  point-cloud, rgbd
Maskfusion
MaskFusion: Real-Time Recognition, Tracking and Reconstruction of Multiple Moving Objects
Stars: ✭ 404 (+116.04%)
Mutual labels:  segmentation, rgbd
Pointclouddatasets
3D point cloud datasets in HDF5 format, containing uniformly sampled 2048 points per shape.
Stars: ✭ 80 (-57.22%)
Mutual labels:  point-cloud, segmentation
Apc Vision Toolbox
MIT-Princeton Vision Toolbox for the Amazon Picking Challenge 2016 - RGB-D ConvNet-based object segmentation and 6D object pose estimation.
Stars: ✭ 277 (+48.13%)
Mutual labels:  segmentation, rgbd
Pointnet
PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Stars: ✭ 3,517 (+1780.75%)
Mutual labels:  point-cloud, segmentation
Record3d
Accompanying library for the Record3D iOS app (https://record3d.app/). Allows you to receive RGBD stream from iOS devices with TrueDepth camera(s).
Stars: ✭ 102 (-45.45%)
Mutual labels:  point-cloud, rgbd
Dgcnn.pytorch
A PyTorch implementation of Dynamic Graph CNN for Learning on Point Clouds (DGCNN)
Stars: ✭ 153 (-18.18%)
Mutual labels:  point-cloud, segmentation

3DGNN for RGB-D segmentation

This is the Pytorch implementation of 3D Graph Neural Networks for RGBD Semantic Segmentation:

Data Preparation

  1. Download NYU_Depth_V2 dataset from here and select scenes and save as ./datasets/data/nyu_depth_v2_labeled.mat
  2. Transfer depth images to hha by yourself from here and save in ./datasets/data/hha/.

Emviroment

Required CUDA (8.0) + pytorch 0.4.1

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