File size: 869 Bytes
f10370d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from .embed import get_embedding
from .utils import load_embeddings, save_embeddings

def register_new_user(face_image, name):
    """
    Registers a new user by generating an embedding and persisting it.
    
    Args:
        face_image (np.array): The input BGR image frame (must contain a face).
        name (str): The name of the user to register.
        
    Returns:
        bool: True on successful registration, False otherwise.
    """
    embedding = get_embedding(face_image)
    
    if embedding is not None:
        data = load_embeddings()
        
        # Store the created embedding and name
        data['embeddings'].append(embedding)
        data['names'].append(name)
        
        # Persist data to {data/embeddings.pkl} [cite: 38]
        return save_embeddings(data)
    
    return False