search

Aradhye2002 / EcoDepth

[CVPR'2024] Official implementation of the paper "ECoDepth: Effective Conditioning of Diffusion Models for Monocular Depth Estimation"
https://ecodepth-iitd.github.io/
MIT License
167 stars 17 forks source link

Predicted depth map only #7

Closed jtsanborn1 closed 7 months ago

jtsanborn1 commented 7 months ago

Is there a way to export a predicted depth map only instead of both image and its depth map side by side? Or/and even to save the grayscale depth map without color palette to the depth map.

Any argument to add to the lines?

frankkim1108 commented 7 months ago

HI, @jtsanborn1 I was trying to do the same thing as you and I think I found a solution.

Add cv2.imwrite('../depth_map.png', depth_map) to line 90 in infer.py file. The writer of this code did viz = np.hstack((img, depth_map)), resulting the output to have both original image and depth map side by side.

Aradhye2002 commented 7 months ago

Yes this is the correct way to proceed. Thanks @frankkim1108.

Closing as completed.