depth-map
depth-map copied to clipboard
It gives a depth map for single image using pre-trained deep learning depth-net model.
trafficstars
depth-map
- 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.