ZF/ViT-Face-Recognition
1
0
Fork 0
mirror of https://github.com/MarcosRodrigoT/ViT-Face-Recognition.git synced 2026-09-09 02:58:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change line length to a max of 88

Browse Source
This commit is contained in:
mrt
2023-09-12 14:25:59 +02:00
committed by Marcos Rodrigo
parent f09624dc0b
commit 3b3ba2e712
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
real_world_occluded_faces_test.py
View File

@@ -22,11 +22,13 @@ def remove_empty_directories():
def get_unique_names(): def get_unique_names():
neutral_items = os.listdir(NEUTRAL_DIR) neutral_items = os.listdir(NEUTRAL_DIR)
masked_items = list(map(lambda x: x.split('_wearing_mask')[0], os.listdir(MASKED_DIR))) masked_items = list(
sunglasses_items = list(map(lambda x: x.split('_wearing_sunglasses')[0], os.listdir(SUNGLASSES_DIR))) map(lambda x: x.split('_wearing_mask')[0], os.listdir(MASKED_DIR))
return sorted(
set(masked_items + neutral_items + sunglasses_items)
) )
sunglasses_items = list(
map(lambda x: x.split('_wearing_sunglasses')[0], os.listdir(SUNGLASSES_DIR))
)
return sorted(set(masked_items + neutral_items + sunglasses_items))
BASE_DIR = '/mnt/Data/mrt/RealWorldOccludedFaces/images' BASE_DIR = '/mnt/Data/mrt/RealWorldOccludedFaces/images'