- Clone this repository:
git clone https://github.com/Abhinandan11/depth-map.git
- Install the dependencies:
pip install -r requirements.txt
-
Download these two files: https://drive.google.com/open?id=1ztQTAoDOzfG9IiZc4rxB5FMJgi40ngvf
https://drive.google.com/open?id=1W7SexKT0Wb8-wqmf23q94tDLmd48wkxK
and move them to models folder.
-
Rename your input image as input.jpg and place it in the main folder.
-
Run predict.py file with command:
python predict.py models/NYU_FCRN.ckpt input.jpg
- It will save two depth maps as final.jpg and finalgray.jpg in the main folder.
Please fork this repository, if you want to use this code and give it a star.