Fix Mean and Base Model weight location and Added Mean Image

neccam · Aug 21, 2018 · de1a769 · de1a769
1 parent 223a6f2
commit de1a769
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+
+\.idea/
diff --git a/Mean/FulFrame_Mean_Image_227x227.npy b/Mean/FulFrame_Mean_Image_227x227.npy
diff --git a/nslt/alexnet.py b/nslt/alexnet.py
@@ -41,7 +41,7 @@ def __init__(self, source, keep_prob, device_str):
         self.device_str = device_str
         self.input = source
         self.keep_prob = keep_prob
-        self.weights_path = '/media/Depo/mofo-cihan/Code/NMT/NMT/bvlc_alexnet.npy'
+        self.weights_path = '../BaseModel/bvlc_alexnet.npy'
 
         # Call the create function to build the computational graph of AlexNet
         self.create()

diff --git a/nslt/utils/iterator_utils.py b/nslt/utils/iterator_utils.py
@@ -74,7 +74,7 @@ def read_video(src, source_reverse):
     video = np.zeros((len(images), 227, 227, 3)).astype(np.float32)
 
     # Cihan_CR: Harcoded Path, Need to Change This
-    mean_image = np.load('/media/Depo/mofo-cihan/Code/NMT/Mean_Calculation/FulFrame_Mean_Image_227x227.npy').astype(np.float32)[..., ::-1]
+    mean_image = np.load('../Mean/FulFrame_Mean_Image_227x227.npy').astype(np.float32)[..., ::-1]
 
     # for each image
     for i in range(0, len(images)):