Update README.md
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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:
|