项目作者: dthanuja

项目描述 :
A CNN based Depth, Optical Flow, Flow Uncertainty and Camera Pose Prediction pipeline
高级语言: Python
项目地址: git://github.com/dthanuja/ENG.git
创建时间: 2019-03-25T02:30:52Z
项目社区:https://github.com/dthanuja/ENG

开源协议:

下载


ENG

ENG: End-to-end neural geometry for robust depth and pose estimation using CNNs


eng

Requirements

The code is tested on CUDA 9.0 + cudNN 7.4.2 + Python 3.5.2 + Tensorflow 1.9

Download the weights (requires wget)

  1. ./download_model.sh

Running

form the root dir of the project run:

  1. python -m test_depth_flow_pose

Run Options

There are some configuration parameters, use the following command to print the help menu

  1. python -m test_depth_flow_pose --help

Citing the Paper(s)

If you found this repository useful please cite the following:

  1. @article{dharmasiri2018eng,
  2. title={ENG: End-to-end Neural Geometry for Robust Depth and Pose Estimation using CNNs},
  3. author={Dharmasiri, Thanuja and Spek, Andrew and Drummond, Tom},
  4. journal={arXiv preprint arXiv:1807.05705},
  5. year={2018}
  6. }

Single Image Depth Prediction Results (KITTI 0-50m)

RMSE (m) Relabs Accuracy (δ) Accuracy (δ2) Accuracy (δ3)

3.284

0.092

90.6%

97.1%

98.9%

Single Image Depth Prediction Results (NYUv2 [using the indoor model] )

RMSE (m) Relabs Accuracy (δ) Accuracy (δ2) Accuracy (δ3)

0.478

0.111

87.2%

97.8%

99.5%

Pose Estimation (KITTI)

Sequence ATE (m) RPE (m) RPE (°)

9

16.55

0.047

0.128

10

9.846

0.039

0.138