utils.py

import numpy as np
import cv2

def preprocess_image(image_array):
    """
    Prépare l'image pour le modèle TFLite.
    - Redimensionne à 128x128 (adapter si ton modèle a une autre taille d'entrée)
    - Normalise entre 0 et 1
    - Ajoute la dimension batch
    """
    image = np.array(image_array, dtype=np.float32)
    image = cv2.resize(image, (128, 128))
    image = image / 255.0
    return image.reshape(1, 128, 128, 3)