Arnaudding001 commited on
Commit
917ff3a
1 Parent(s): 6add330

Update rft_demo.py

Browse files
Files changed (1) hide show
  1. rft_demo.py +4 -4
rft_demo.py CHANGED
@@ -1,5 +1,5 @@
1
  import sys
2
- sys.path.append('core')
3
 
4
  import argparse
5
  import os
@@ -9,9 +9,9 @@ import numpy as np
9
  import torch
10
  from PIL import Image
11
 
12
- from raft import RAFT
13
- from utils import flow_viz
14
- from utils.utils import InputPadder
15
 
16
 
17
 
 
1
  import sys
2
+ #sys.path.append('core')
3
 
4
  import argparse
5
  import os
 
9
  import torch
10
  from PIL import Image
11
 
12
+ from raft_core_raft import RAFT
13
+ from raft_core_utils import flow_viz
14
+ from raft_core_utils_utils import InputPadder
15
 
16
 
17