XiWJ / GeoNet

GeoNet: Geometric Neural Network for Joint Depth and Surface Normal Estimation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GeoNet

GeoNet: Geometric Neural Network for Joint Depth and Surface Normal Estimation

GeoNet++: Interative Geometric Neural network with Edge-aware Refinement Joint Depth and Surface Normal Estimation. (In preparation)

Setup

Requirement

Required python libraries: Tensorflow (>=1.2) + Scipy + Numpy + Scipy + OpenCV.

Tested in Ubuntu + Intel i7 CPU + Nvidia Titan X (Pascal) with Cuda (>=8.0) and CuDNN (>=5.0). CPU mode should also work with minor changes.

Inference

  1. Download pretrained model data, initialization model, and trained model from "https://drive.google.com/open?id=1o2t8735acVf2cLSCS6URkNViOB7mdb-Q"

  2. tar xvzf GeoNet.tar.gz

  3. Merge files into the Repo according to the file name.

  4. Run 'code.py'

  5. Evaluation: cd eval & run 'test_depth.m' for depth evaluation and run 'test_norm.py' for normal evaluation.

Training

Training code already included. Detailed data preparation & running procedure will be updated soon.

About

GeoNet: Geometric Neural Network for Joint Depth and Surface Normal Estimation


Languages

Language:Python 88.9%Language:MATLAB 9.9%Language:Limbo 1.2%