Anonymous-sub commited on
Commit
da6c999
1 Parent(s): deab087

Update flow/flow_utils.py

Browse files
Files changed (1) hide show
  1. 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):