ankuraditya commited on
Commit
89dc03f
Β·
verified Β·
1 Parent(s): 6294a17

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -1
README.md CHANGED
@@ -46,6 +46,28 @@ umass-lass/ReVo
46
  └── dcvcrt/
47
  β”œβ”€β”€ dcvcrt_rgb.pth
48
  β”œβ”€β”€ dcvcrt_depth.pth
 
49
 
 
50
 
51
- # How to Use
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
46
  └── dcvcrt/
47
  β”œβ”€β”€ dcvcrt_rgb.pth
48
  β”œβ”€β”€ dcvcrt_depth.pth
49
+ ```
50
 
51
+ # How to Use?
52
 
53
+ ```python
54
+ from huggingface_hub import hf_hub_download
55
+ import torch
56
+
57
+ # Define the specific codec and modality you want to load
58
+ codec = "h265" # Options: 'h264', 'h265', 'dcvcrt'
59
+ modality = "depth" # Options: 'rgb', 'depth'
60
+
61
+ # Download the specific checkpoint
62
+ checkpoint_path = hf_hub_download(
63
+ repo_id="umass-lass/ReVo",
64
+ filename=f"{codec}/{codec}_{modality}.pth"
65
+ )
66
+
67
+ # Load the weights into your PyTorch model
68
+ # model.load_state_dict(torch.load(checkpoint_path))
69
+ print(f"Successfully downloaded to: {checkpoint_path}")
70
+ ```
71
+
72
+ # Citation
73
+ (Coming Soon)