superxiaoying / SISNet

Semantic Scene Completion via Integrating Instances and Scene in-the-Loop (CVPR 2021)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SISNet

Semantic Scene Completion via Integrating Instances and Scene in-the-Loop (CVPR 2021)

In this repository, we provide SISNet model implementation (with Pytorch) as well as data preparation, training and evaluation scripts on NYU Depth V2, NYUCAD and SUNCG_ RGBD.

Installation The code is tested with Ubuntu 18.04, Pytorch v1.4.0, CUDA 10.0 and cuDNN v7.4.

conda install pytorch==1.4.0

About

Semantic Scene Completion via Integrating Instances and Scene in-the-Loop (CVPR 2021)