cr fixes

load_test_data.py

@@ -4,11 +4,12 @@ from skimage.io import imread
 import cv2 as cv
 from pathlib import Path
 
+
 def load_data(input_dir, newSize=(64,64)):
     image_path = Path(input_dir)
     file_names = os.listdir(image_path)
     categories_name = []
-    categories_count=[]
+    categories_count = []
     count = 0
     n = file_names[0]
     for name in file_names:
@@ -22,7 +23,7 @@ def load_data(input_dir, newSize=(64,64)):
             categories_name.append(name)
     categories_count.append(count)
     test_img = []
-    labels=[]
+    labels = []
 
     for n in file_names:
         p = image_path / n
@@ -32,15 +33,9 @@ def load_data(input_dir, newSize=(64,64)):
         test_img.append(img)
         labels.append(n)
 
-    X={}
+    X = {}
     X["values"] = np.array(test_img)
-    X["categories_name"] = categories_name
-    X["categories_count"] = categories_count
-    X["labels"]=labels
-    return X
-
-data = load_data('test_set')
-print(data['categories_name'])
-print(data['categories_count'])
-print(data['labels'])
-print(list(data["values"]))
+    X["name"] = categories_name
+    X["names_count"] = categories_count
+    X["labels"] = labels
+    return X