Hybrid DenseNet-based CNN for Single Image Depth Estimation
This repository contains an updated version of the CNN from my repository DE_resnet_unet_hyb. The ResNet backbone has been replaced with a DenseNet169.
Requirements
The code was tested with:
python 3.5 and 3.6
pytorch (and torchvision) 1.3.0
opencv-python 3.4.3
matplotlib 2.2.3
numpy 1.15.4
Guide
Predicting the depth of an arbitrary image:
python3 predict_img.py -i <path_to_image>
Evalutation
Quantitative results on the NYU depth v2 test set: