search

isl-org / ZoeDepth

Metric depth estimation from a single image
MIT License
2.25k stars 207 forks source link

Train on custom dataset #24

Open ecatanzani opened 1 year ago

ecatanzani commented 1 year ago

Hi, I don't find a part of the documentation talking about training the model on a custom set of images with depth associated. Can you please provide me some docs?

kwea123 commented 1 year ago

I recommend not. I successfully refactored the code a lot to just integrate a single custom data loader, then found out the original training doesn't even always work. I wasted two days.

ecatanzani commented 1 year ago

Can you share the code in order to understand how to create a custom dataset? At least I can give a try to integrate mine successfully

jubileekr commented 1 year ago

Found this, it might help? https://medium.com/@bhaskarbose1998/monocular-depth-estimation-using-zoedepth-our-experience-42fa5974cb59

ecatanzani commented 1 year ago

I've read your reference. It refers to RDE while I was wandering something related to MDE... any updates in that sense?