s444498/wko_anime-face-similarity
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

cr fixes

Browse Source
This commit is contained in:
Mateusz Tylka 2023-01-29 23:18:50 +01:00
parent dac16b130b
commit 0622beddea
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
load_test_data.py
View File

@ -4,6 +4,7 @@ from skimage.io import imread
import cv2 as cv import cv2 as cv
from pathlib import Path from pathlib import Path
def load_data(input_dir, newSize=(64,64)): def load_data(input_dir, newSize=(64,64)):
image_path = Path(input_dir) image_path = Path(input_dir)
file_names = os.listdir(image_path) file_names = os.listdir(image_path)
@ -34,13 +35,7 @@ def load_data(input_dir, newSize=(64,64)):
X = {} X = {}
X["values"] = np.array(test_img) X["values"] = np.array(test_img)
X["categories_name"] = categories_name X["name"] = categories_name
X["categories_count"] = categories_count X["names_count"] = categories_count
X["labels"] = labels X["labels"] = labels
return X return X
data = load_data('test_set')
print(data['categories_name'])
print(data['categories_count'])
print(data['labels'])
print(list(data["values"]))