Views
No views yet
1from dkt.pipelines.pipelines import DKTPipeline
2import os
3from tools.common_utils import save_video
4
5pipe = DKTPipeline()
6demo_path = 'examples/1.mp4'
7prediction = pipe(demo_path)
8save_dir = 'logs'
9os.makedirs(save_dir, exist_ok=True)
10output_path = os.path.join(save_dir, 'demo.mp4')
11save_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}