create test set #2
@ -4,6 +4,7 @@ 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)
|
||||
@ -34,13 +35,7 @@ def load_data(input_dir, newSize=(64,64)):
|
||||
|
||||
|
||||
X = {}
|
||||
X["values"] = np.array(test_img)
|
||||
X["categories_name"] = categories_name
|
||||
X["categories_count"] = categories_count
|
||||
X["name"] = categories_name
|
||||
X["names_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"]))
|
Loading…
Reference in New Issue
Block a user
Hm, nie lepiej byłoby żeby ta struktura wyglądała w ten sposób?