INNER CODE UNIT · Python

tflite_metadata

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

def tflite_metadata(path: Path) -> dict | None:
    """Return embedded Ultralytics TFLite metadata when present."""
    try:
        with zipfile.ZipFile(path) as zf:
            infos = [
                info for info in zf.infolist() if info.filename in {"metadata.json", "TFLITE_ULTRALYTICS_METADATA.json"}
            ]
            if infos:
                return json.loads(zf.read(infos[-1]))
    except Exception:
        return None
    return None


def task_names(task_name: str, suffix: str) -> dict[int, str]:
    """Return class names for the representative model in a task family."""
    if task_name not in _TASK_NAMES_CACHE:
        from ultralytics import YOLO

View source record →