roop/FaceSet.py

import numpy as np


class FaceSet:
    faces = []
    ref_images = []
    embedding_average = "None"
    embeddings_backup = None

    def __init__(self):
        self.faces = []
        self.ref_images = []
        self.embeddings_backup = None

    def AverageEmbeddings(self):
        if len(self.faces) > 1 and self.embeddings_backup is None:
            self.embeddings_backup = self.faces[0]["embedding"]
            embeddings = [face.embedding for face in self.faces]

            self.faces[0]["embedding"] = np.mean(embeddings, axis=0)
            # try median too?