search

shariqfarooq123 / AdaBins

Official implementation of Adabins: Depth Estimation using adaptive bins
GNU General Public License v3.0
725 stars 156 forks source link

How to output depth to image ? #55

Closed Entretoize closed 2 years ago

Entretoize commented 2 years ago

You give an example of how to use but it return bin_centes and predicted_depth, what to do with that ? How to export to an image ?

Thanks

Entretoize commented 2 years ago

I tried using predict_dir() but Image.fromarray doesn't seem to handle that and gives TypeError: Cannot handle this data type: (1, 1, 480, 640), <u2

shariqfarooq123 commented 2 years ago

This TypeError has been fixed