shahabty / depth-estimation

Edge Aware Monocular Depth Estimation

Edge Aware Monocular Depth Estimation

This repo contains an implementation of Resnext baseline for depth estimation in Pytorch.

Prerequisites

Download NYUV2 dataset and pre-trained models. Then follow the installation instruction below:

Installation

1. Install Anaconda3

2. Run the following commands to create conda environment and install all dependencies:

username@PC:~$ conda env create -f environment.yml
username@PC:~$ conda activate mde

3. You can modify cfg, train_args and test_args to change training and testing settings.

Acknowledgments

This implementation is partially based on official implementation of Enforcing geometric constraints of virtual normal for depth prediction..