Skip to content
3D Graph Neural Networks for RGBD Semantic Segmentation
Python
Branch: master
Clone or download
Latest commit b5e0188 Mar 16, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
datasets
.gitattributes Initial commit Nov 25, 2018
.gitignore Initial commit Nov 25, 2018
LICENSE Create LICENSE Mar 4, 2019
README.md
config.py configuration Jan 24, 2019
draw_net.py
models.py
run.py train model Mar 14, 2019

README.md

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

You can’t perform that action at this time.