add paths before import

zhangkaiitugithub · Jun 1, 2017 · 0ebecf3 · 0ebecf3
1 parent ed4f43a
commit 0ebecf3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/examples/create_dataset_and_use_readerop.py b/examples/create_dataset_and_use_readerop.py
@@ -6,10 +6,10 @@
 import numpy as np
 from PIL import Image
 import h5py
-from depthmotionnet.dataset_tools import *
 
 examples_dir = os.path.dirname(__file__)
 sys.path.insert(0, os.path.join(examples_dir, '..', 'python'))
+from depthmotionnet.dataset_tools import *
 
 
 # intrinsics supported by DeMoN

diff --git a/examples/evaluation.py b/examples/evaluation.py
@@ -15,19 +15,17 @@
 import h5py
 import xarray
 import numpy as np
-from depthmotionnet.datareader import *
-from depthmotionnet.networks_original import *
-from depthmotionnet.helpers import convert_NCHW_to_NHWC, convert_NHWC_to_NCHW
-from depthmotionnet.evaluation import *
 import lmbspecialops as sops
 import tensorflow as tf
 
-
-
 examples_dir = os.path.dirname(__file__)
 weights_dir = os.path.join(examples_dir,'..','weights')
 sys.path.insert(0, os.path.join(examples_dir, '..', 'python'))
 
+from depthmotionnet.datareader import *
+from depthmotionnet.networks_original import *
+from depthmotionnet.helpers import convert_NCHW_to_NHWC, convert_NHWC_to_NCHW
+from depthmotionnet.evaluation import *
 
 
 def create_ground_truth_file(dataset, dataset_dir):