Thanks to visit codestin.com
Credit goes to github.com

Skip to content

code for "FiffDepth: Feed-forward Transformation of Diffusion-Based Generators for Detailed Depth Estimation"

Notifications You must be signed in to change notification settings

yunpeng1998/FiffDepth

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FiffDepth: Feed-forward Transformation of Diffusion-Based Generators for Detailed Depth Estimation

We propose an efficient MDE approach named FiffDepth. The key feature of FiffDepth is its use of diffusion priors. It transforms diffusion-based image generators into a feed-forward architecture for detailed depth estimation. FiffDepth preserves key generative features and integrates the strong generalization capabilities of models like DINOv2. Through benchmark evaluations, we demonstrate that FiffDepth achieves exceptional accuracy, stability, and fine-grained detail, offering significant improvements in MDE performance against state-of-the-art MDE approaches.

Overview

Environment Setup

To create a Python virtual environment named fiffdepth under the ./venv directory and install all dependencies from requirements.txt, run:

python3 -m venv ./venv/fiffdepth
source ./envs/fiffdepth/bin/activate
pip install -r requirements.txt

Checkpoint Preparation

Download all checkpoints from here and place them under:

./checkpoints/fiffdepth

This is a retrained checkpoint. Since it was trained with different real image data, its performance may not be optimal.

Inference

python run_direct.py --checkpoint './checkpoints/fiffdepth' --sd_ckpt ./checkpoints/fiffdepth' \ --input_rgb_dir input/images --output_dir output/depth

Citation

@inproceedings{bai2025fiffdepth,
  title={Fiffdepth: Feed-forward transformation of diffusion-based generators for detailed depth estimation},
  author={Bai, Yunpeng and Huang, Qixing},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={6023--6033},
  year={2025}
}

About

code for "FiffDepth: Feed-forward Transformation of Diffusion-Based Generators for Detailed Depth Estimation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages