from pathlib import Path


if __name__ == "__main__":
    from ultralytics import YOLO
    default_model_path = Path(__file__).resolve().parent.parent / "yolo/models/yolov8n-face.pt"
    default_model_path = str(default_model_path)  # Convertir en chaîne de caractères pour l'API YOLO
    # Charger votre modèle YOLO customisé pour les visages
    model = YOLO(default_model_path)
    # Exporter au format Edge TPU (gère la quantification et la compilation automatiquement)
    model.export(format="edgetpu")