3DGNN for RGB-D segmentation

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

Required CUDA (8.0) + pytorch 0.4.1

Name	Name	Last commit message	Last commit date
Latest commit yanx27 Update README.md Mar 16, 2019 b5e0188 · Mar 16, 2019 History 34 Commits
datasets	datasets	Load dataset	Jan 24, 2019
.gitattributes	.gitattributes	Initial commit	Nov 25, 2018
.gitignore	.gitignore	Initial commit	Nov 25, 2018
LICENSE	LICENSE	Create LICENSE	Mar 4, 2019
README.md	README.md	Update README.md	Mar 16, 2019
config.py	config.py	configuration	Jan 24, 2019
draw_net.py	draw_net.py	show_model	Jan 24, 2019
models.py	models.py	Update models.py	Mar 14, 2019
run.py	run.py	train model	Mar 14, 2019

Provide feedback