yunzhong-hou commited on
Commit
4ea9e23
·
verified ·
1 Parent(s): 1ff5f27

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -1
README.md CHANGED
@@ -31,7 +31,20 @@ Abstract: *This study seeks to automate camera movement control for filming exis
31
  Please refer to our [GitHub repo](https://github.com/hou-yz/dvgformer) for detailed installation steps.
32
 
33
  ## Usage
34
- Download the model first and put it under the `checkpoints/` directory.
 
 
 
 
 
 
 
 
 
 
 
 
 
35
 
36
  ## Citation
37
  If you find this project useful, please consider citing:
 
31
  Please refer to our [GitHub repo](https://github.com/hou-yz/dvgformer) for detailed installation steps.
32
 
33
  ## Usage
34
+ Assuming you have followed the Installation Steps from [GitHub repo](https://github.com/hou-yz/dvgformer), you can use the following code to load the model checkpoint
35
+ ```python
36
+ import torch
37
+ from src.models import DVGFormerModel
38
+
39
+ model = DVGFormerModel.from_pretrained(
40
+ 'yunzhong-hou/DVGFormer'
41
+ ).to('cuda').to(torch.bfloat16)
42
+ ```
43
+
44
+ For blender evaluation, you can run the following script from the GitHub repo.
45
+ ```sh
46
+ python blender_eval.py
47
+ ```
48
 
49
  ## Citation
50
  If you find this project useful, please consider citing: