Files
ViT-Face-Recognition/data_generator.py