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

19
load_test_data.py
View File

@ -4,11 +4,12 @@ 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)
categories_name = [] categories_name = []
categories_count=[] categories_count = []
count = 0 count = 0
n = file_names[0] n = file_names[0]
for name in file_names: for name in file_names:
@ -22,7 +23,7 @@ def load_data(input_dir, newSize=(64,64)):
categories_name.append(name) categories_name.append(name)
categories_count.append(count) categories_count.append(count)
test_img = [] test_img = []
labels=[] labels = []
for n in file_names: for n in file_names:
p = image_path / n p = image_path / n
@ -32,15 +33,9 @@ def load_data(input_dir, newSize=(64,64)):
test_img.append(img) test_img.append(img)
labels.append(n) labels.append(n)
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"]))