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

create test set #2

Merged
s444409 merged 3 commits from test_set into main 2023-01-31 19:54:54 +01:00
Conversation 0 Commits 3 Files Changed 51 +8 -13
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 0622beddea - Show all commits

21
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 = []
s444409 commented 2023-01-29 23:01:48 +01:00
Review
Copy Link

categories_count = []

`categories_count = []`
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 = []
s444409 commented 2023-01-29 23:01:30 +01:00
Review
Copy Link

labels = []

`labels = []`
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)
s444409 commented 2023-01-29 23:05:59 +01:00
Review
Copy Link

Hm, nie lepiej byłoby żeby ta struktura wyglądała w ten sposób?

[
	{
    	"name": "plik.jpg"
        "values": <np.ndarray>
    },
    ...
]
Hm, nie lepiej byłoby żeby ta struktura wyglądała w ten sposób? ```json [ { "name": "plik.jpg" "values": <np.ndarray> }, ... ] ```
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"]))