shaocong commited on
Commit
09e6791
·
1 Parent(s): 937cb83

first commit

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ TransPhy3D_cleargrasp_HISS_DREDS_DREDS_14B_depth_70k_lora.safetensors filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,3 +1,26 @@
1
  ---
2
  license: apache-2.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
  ---
4
+ # Model Card for DKT
5
+
6
+ This repository contains the weights of `Diffusion Knows Transparency: Repurposing Video Diffusion for Transparent Object Depth and Normal Estimation`
7
+
8
+ ## Usage
9
+
10
+ See the Github repository: [DKT](https://github.com/Daniellli/DKT) regarding installation instructions.
11
+
12
+ The model can then be used as follows:
13
+
14
+ ```python
15
+ from dkt.pipelines.pipelines import DKTPipeline
16
+ import os
17
+ from tools.common_utils import save_video
18
+
19
+ pipe = DKTPipeline()
20
+ demo_path = 'examples/1.mp4'
21
+ prediction = pipe(demo_path)
22
+ save_dir = 'logs'
23
+ os.makedirs(save_dir, exist_ok=True)
24
+ output_path = os.path.join(save_dir, 'demo.mp4')
25
+ save_video(prediction['colored_depth_map'], output_path, fps=25)
26
+ ```
TransPhy3D_cleargrasp_HISS_DREDS_DREDS_14B_depth_70k_lora.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5d2f8773d586d2186adaf51c1fea13d18d79c7269a318af9a2e627cd6c67a1c7
3
+ size 1226928552