ZF/ViT-Face-Recognition
1
0
Fork 0
mirror of https://github.com/MarcosRodrigoT/ViT-Face-Recognition.git synced 2025-12-30 08:02:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add file paths to the dictionary containing all the data

Browse Source
This commit is contained in:
Marcos Rodrigo
2023-09-21 12:16:48 +02:00
parent 028a0798a0
commit 9a9e86e56d
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
real_world_occluded_faces_test.py
View File

@@ -79,7 +79,9 @@ def get_data():
for file in sorted(os.listdir(f"{NEUTRAL_DIR}/{folder}")):
person_name = folder
file_name = file.split('.jpg')[0]
file_path = os.path.join(f"{NEUTRAL_DIR}/{folder}/{file}")
data[person_name]['Neutral'][file_name] = {
'file': file_path,
'embeddings': {
'vit': None,
'resnet': None,
@@ -93,7 +95,9 @@ def get_data():
for file in sorted(os.listdir(f"{MASKED_DIR}/{folder}")):
person_name = folder.split('_wearing_mask')[0]
file_name = file.split('.jpg')[0]
file_path = os.path.join(f"{MASKED_DIR}/{folder}/{file}")
data[person_name]['Masked'][file_name] = {
'file': file_path,
'embeddings': {
'vit': None,
'resnet': None,
@@ -107,7 +111,9 @@ def get_data():
for file in sorted(os.listdir(f"{SUNGLASSES_DIR}/{folder}")):
person_name = folder.split('_wearing_sunglasses')[0]
file_name = file.split('.jpg')[0]
file_path = os.path.join(f"{MASKED_DIR}/{folder}/{file}")
data[person_name]['Sunglasses'][file_name] = {
'file': file_path,
'embeddings': {
'vit': None,
'resnet': None,