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

Get embeddings for mugshot and surveillance images

Browse Source
This commit is contained in:
mrt
2023-09-14 15:43:47 +02:00
parent a760a794fb
commit 312cc07f75
1 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
scface_test.py
View File

@@ -170,7 +170,6 @@ try:
mugshot_data, surveillance_data = pickle.load(file)
except FileNotFoundError:
for person in mugshot_data.keys():
print(f'##### Person {person} #####')
img = preprocess_image(mugshot_data[person]['file'])
embeddings_vit = vit_model(img).numpy()
@@ -187,5 +186,24 @@ except FileNotFoundError:
mugshot_data[person]['embeddings']['mobilenet'] = embeddings1_mobilenet
mugshot_data[person]['embeddings']['efficientnet'] = embeddings1_efficientnet
# for person in surveillance_data.keys():
# pass
for person in surveillance_data.keys():
for file in surveillance_data[person].keys():
img = preprocess_image(surveillance_data[person][file]['file'])
embeddings_vit = vit_model(img).numpy()
embeddings1_resnet = resnet50_model(img).numpy()
embeddings1_vgg16 = vgg16_model(img).numpy()
embeddings1_inception = inception_model(img).numpy()
embeddings1_mobilenet = mobilenet_model(img).numpy()
embeddings1_efficientnet = efficientnetB0_model(img).numpy()
surveillance_data[person][file]['embeddings']['vit'] = embeddings_vit
surveillance_data[person][file]['embeddings']['resnet'] = embeddings1_resnet
surveillance_data[person][file]['embeddings']['vgg'] = embeddings1_vgg16
surveillance_data[person][file]['embeddings']['inception'] = embeddings1_inception
surveillance_data[person][file]['embeddings']['mobilenet'] = embeddings1_mobilenet
surveillance_data[person][file]['embeddings']['efficientnet'] = embeddings1_efficientnet
with open('./saved_results/Tests/SCface/embeddings.pickle', 'wb') as file:
data = (mugshot_data, surveillance_data)
pickle.dump(data, file)