embeddings_generation.py

class FaceEmbedding:
    def __init__(self, transform, resnet):
        self.transform = transform
        self.resnet = resnet

    def calculate_face_embedding(self, image, box):
        face = image.crop(box)
        face = face.convert("RGB")
        face = self.transform(face)
        face = face.unsqueeze(0)
        face_embedding = self.resnet(face)
        return face_embedding