isl-org / ZoeDepth

Metric depth estimation from a single image
MIT License
2.36k stars 216 forks source link

Quick question about modifying the loss function #99

Open 7yzx opened 10 months ago

7yzx commented 10 months ago

Thank you for your great work ! However, as a beginner, I'm currently looking to obtain a depth map that emphasizes edge information more. I am considering modifying the loss function, such as adding additional gradient penalties and normal vector penalties. Unfortunately, my current hardware only allows for fine-tuning the metric head. In your opinion, is it feasible to achieve this by modifying the loss function within the constraints of fine-tuning the metric head? I greatly appreciate your help.