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

Modified paths to be relative to the project structure

Browse Source
This commit is contained in:
Marcos
2023-03-02 15:32:34 +01:00
parent aab4b5e73a
commit 48af9d5a61
3 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
data_generator.py
View File

@@ -9,8 +9,8 @@ for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
vggface2_train = './datasets/VGG-Face2/data/train' # Total nº files: 3141890 'Z:/Datasets/VGG-Face2/data/train'
vggface2_test = './datasets/VGG-Face2/data/test' # Total nº files: 169396 'Z:/Datasets/VGG-Face2/data/test'
vggface2_train = './datasets/VGG-Face2/data/train' # Total nº files: 3141890
vggface2_test = './datasets/VGG-Face2/data/test' # Total nº files: 169396
dataset_path = vggface2_train
labels_list = [tf.constant(x) for x in sorted(os.listdir(dataset_path))]

2
lfw_test.py
View File

@@ -121,7 +121,7 @@ CREATE DATASET WITH PAIRS
"""
pairs = {}
with open('Z:/Datasets/LFW/pairs.txt', 'r') as file:
with open('./datasets/LFW/pairs.txt', 'r') as file:
for idx, line in enumerate(file):
line_aux = line.split('\n')[0].split('\t')
if len(line_aux) < 4:

16
gti-face_test.py → upm-gti-face_test.py
View File

@@ -474,25 +474,25 @@ GTI-FACE DATASET
""" INDOOR """
# Not Masked
GALLERY_I_N = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Indoor/Gallery/Not Masked/'
PROBE_I_N = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Indoor/Camera/Not Masked/Frames/Curated/'
GALLERY_I_N = './datasets/UPM-GTI-Face/Indoor/Gallery/Not Masked/'
PROBE_I_N = './datasets/UPM-GTI-Face/Indoor/Camera/Not Masked/Frames/Curated/'
GALLERY_SUBJECTS_I_N = ['Andres', 'Cristina', 'DaniB', 'DaniF', 'Diego', 'Ester', 'German', 'Isa', 'Marcos', 'Narciso', 'Pablo']
PROBE_SUBJECTS_I_N = ['Andres', 'Cristina', 'DaniB', 'DaniF', 'Diego', 'Ester', 'German', 'Isa', 'Marcos_2', 'Narciso', 'Pablo']
# Masked
GALLERY_I_M = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Indoor/Gallery/Masked/'
PROBE_I_M = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Indoor/Camera/Masked/Frames/Curated/'
GALLERY_I_M = './datasets/UPM-GTI-Face/Indoor/Gallery/Masked/'
PROBE_I_M = './datasets/UPM-GTI-Face/Indoor/Camera/Masked/Frames/Curated/'
GALLERY_SUBJECTS_I_M = ['Andres', 'Cristina', 'DaniB', 'DaniF', 'Diego', 'Ester', 'German', 'Isa', 'Marcos', 'Narciso', 'Pablo']
PROBE_SUBJECTS_I_M = ['Andres', 'Cristina', 'DaniB', 'DaniF', 'Diego', 'Ester', 'German', 'Isa', 'Marcos_2', 'Narciso', 'Pablo']
""" OUTDOOR """
# Not Masked
GALLERY_O_N = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Outdoor/Gallery/Not Masked/'
PROBE_O_N = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Outdoor/Camera/Not Masked/Frames/Curated/'
GALLERY_O_N = './datasets/UPM-GTI-Face/Outdoor/Gallery/Not Masked/'
PROBE_O_N = './datasets/UPM-GTI-Face/Outdoor/Camera/Not Masked/Frames/Curated/'
GALLERY_SUBJECTS_O_N = ['Andres', 'Cristina', 'DaniB', 'DaniF_2', 'Diego', 'Ester', 'German', 'Isa_2', 'Marcos', 'Narciso', 'Pablo']
PROBE_SUBJECTS_O_N = ['Andres', 'Cristina', 'DaniB', 'DaniF_2', 'Diego', 'Ester', 'German', 'Isa_2', 'Marcos', 'Narciso', 'Pablo']
# Masked
GALLERY_O_M = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Outdoor/Gallery/Masked/'
PROBE_O_M = 'Y:/Doctorado UPM/Paper ICIP 2022/ICIP DDBB/Outdoor/Camera/Masked/Frames/Curated/'
GALLERY_O_M = './datasets/UPM-GTI-Face/Outdoor/Gallery/Masked/'
PROBE_O_M = './datasets/UPM-GTI-Face/Outdoor/Camera/Masked/Frames/Curated/'
GALLERY_SUBJECTS_O_M = ['Andres', 'Cristina', 'DaniB', 'DaniF_2', 'Diego', 'Ester', 'German', 'Isa_2', 'Marcos', 'Narciso', 'Pablo']
PROBE_SUBJECTS_O_M = ['Andres', 'Cristina', 'DaniB', 'DaniF_2', 'Diego', 'Ester', 'German', 'Isa_2', 'Marcos', 'Narciso', 'Pablo']