INNER CODE UNIT · Python

append_tflite_metadata

ultralytics/yolo-flutter-app · scripts/export-tflite-models.py:125

def append_tflite_metadata(path: Path, model_id: str, task_name: str, task: TaskSpec) -> None:
    """Append Ultralytics metadata to a TFLite model archive."""
    from ultralytics import __version__ as ultralytics_version

    metadata = {
        "description": f"Ultralytics {model_id} w8a32 LiteRT model",
        "author": "Ultralytics",
        "date": time.strftime("%Y-%m-%d"),
        "version": ultralytics_version,
        "task": task_name,
        "batch": 1,
        "imgsz": [task.imgsz, task.imgsz],
        "names": task_names(task_name, task.suffix),
        "channels": 3,
        "stride": 32,
        "format": "litert",
        "int8": False,
        "end2end": False,

View source record →