Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
da6c999
1
Parent(s):
deab087
Update flow/flow_utils.py
Browse files- flow/flow_utils.py +6 -0
flow/flow_utils.py
CHANGED
|
@@ -12,7 +12,13 @@ sys.path.insert(0, gmflow_dir)
|
|
| 12 |
from gmflow.gmflow import GMFlow # noqa: E702 E402 F401
|
| 13 |
from utils.utils import InputPadder # noqa: E702 E402
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
def coords_grid(b, h, w, homogeneous=False, device=None):
|
|
|
|
| 12 |
from gmflow.gmflow import GMFlow # noqa: E702 E402 F401
|
| 13 |
from utils.utils import InputPadder # noqa: E702 E402
|
| 14 |
|
| 15 |
+
import huggingface_hub
|
| 16 |
+
|
| 17 |
+
repo_name = 'Anonymous-sub/Rerender'
|
| 18 |
+
|
| 19 |
global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 20 |
+
gmflow_path = huggingface_hub.hf_hub_download(
|
| 21 |
+
repo_name, 'models/gmflow_sintel-0c07dcb3.pth', local_dir='models')
|
| 22 |
|
| 23 |
|
| 24 |
def coords_grid(b, h, w, homogeneous=False, device=None):
|