How to run on Custom data?

princeton-vl / OGNI-DC

[ECCV24] official code for "OGNI-DC: Robust Depth Completion with Optimization-Guided Neural Iterations"

BSD 3-Clause "New" or "Revised" License

45 stars 1 forks source link

How to run on Custom data? #2

Closed aniket-gupta1 closed 2 months ago

aniket-gupta1 commented 3 months ago

Hi, thanks for your amazing work.

Can you provide some pointers as to what's the easiest way to run your model on custom data?

zuoym15 commented 2 months ago

Hi, thank you for your interest in our work!

I would recommend modifying based on the DDAD dataloader (src/data/ddad.py). You just need the RGB image and the sparse depth map (in meters) and can set the intrinsic and the gt to be zeros.

Feel free to post an update if you run into any specific issues.