facenet-pytorch
torch
datasets