Views
No views yet
1from dkt.pipelines.pipelines import DKTPipeline
2import os
3from tools.common_utils import save_video
4
5# Initialize the pipeline
6pipe = DKTPipeline()
7
8# Define input path
9demo_path = 'examples/1.mp4'
10
11# Run inference
12prediction = pipe(demo_path)
13
14# Save result
15save_dir = 'logs'
16os.makedirs(save_dir, exist_ok=True)
17output_path = os.path.join(save_dir, 'demo.mp4')
18save_video(prediction['colored_depth_map'], output_path, fps=25)1@article{dkt2025,
2 title = {Diffusion Knows Transparency: Repurposing Video Diffusion for Transparent Object Depth and Normal Estimation},
3 author = {Shaocong Xu and Songlin Wei and Qizhe Wei and Zheng Geng and Hong Li and Licheng Shen and Qianpu Sun and Shu Han and Bin Ma and Bohan Li and Chongjie Ye and Yuhang Zheng and Nan Wang and Saining Zhang and Hao Zhao},
4 journal = {https://arxiv.org/abs/2512.23705},
5 year = {2025}
6}