Fix Issue #1

src/data_processing.py

@@ -37,6 +37,8 @@ def store_data(filenames, input_path, output_path):
             print(error_message)
 
 def main():
+    if not os.path.exists(sys.argv[1]): 
+        raise ValueError("The image files doesn't Exists. Please review your file path")
     dir = sys.argv[1]
     input_filepaths = get_file_names(dir)
     output_filepath = sys.argv[2]

src/image_alignment.py

@@ -1,5 +1,5 @@
 import sys
-
+import os.path
 import cv2
 import numpy as np
 import image_shapes as shapes
@@ -90,6 +90,8 @@ def align_images(petal_img, vein_img, raw_vein=True):
 def main():
     if (len(sys.argv) < 3):
         raise ValueError("Usage: image_alignment.py <petal image> <vein image>")
+    if not (os.path.exists(sys.argv[1]) and os.path.exists(sys.argv[2])): 
+        raise ValueError("One of the image files doesn't Exists. Please review your file paths \n Usage: image.alignment.py <correct image path> <correct image path>")  
     petal_image = cv2.imread(sys.argv[1])
     vein_image = cv2.imread(sys.argv[2], cv2.IMREAD_GRAYSCALE)
     (petal_shape, vein_aligned) = align_images(petal_image, vein_image)

src/imgrab.py

@@ -1,10 +1,13 @@
 import principal_component as p
 import sys
+import os.path
 import cv2
 from skimage import io
 
 
 def main():
+    if not os.path.exists(sys.argv[1]): 
+        raise ValueError("The image files doesn't Exists. Please review your file path")
     image = cv2.imread(sys.argv[1])
     im = vein_image = cv2.imread(sys.argv[1], cv2.IMREAD_GRAYSCALE)
     i = p.pca_to_grey(image)

src/principal_component.py

@@ -1,5 +1,6 @@
 import sys
 import cv2
+import os.path
 import matplotlib.pyplot as plt
 import numpy as np
 import image_shapes as shapes
@@ -59,11 +60,13 @@ def create_point_cloud(image):
     if len(sys.argv) < 2:
         print("Standalone usage: python principal_component.py input_filename.jpg [output_filename.jpg]")
         sys.exit(1)
+    if not os.path.exists(sys.argv[1]): 
+        raise ValueError("One of the image files doesn't Exists. Please review your file paths") 
     petal_image = cv2.imread(sys.argv[1])
     petal_shape = shapes.get_petal_shape(petal_image)
     result = pca_to_grey(petal_image, petal_shape, True)
     if len(sys.argv) > 2:
         cv2.imwrite(sys.argv[2], result)
     else:
         io.imshow(result)
-        io.show()
+        io.show()

src/spot_detection.py

@@ -1,5 +1,5 @@
 import sys
-
+import os.path
 import matplotlib.pyplot as plt
 import numpy as np
 from sklearn.mixture import BayesianGaussianMixture
@@ -56,6 +56,8 @@ def overlay_spotting_events_on_image(image, model):
     plt.show()
 
 def main():
+    if not os.path.exists(sys.argv[1]): 
+        raise ValueError("The image files doesn't Exists. Please review your file path")
     image = imread(sys.argv[1])
     image = iu.resize_image(image, 400)