INNER CODE UNIT · Python

task_names

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

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

        weights_name = f"yolo26m{suffix}.pt"
        weights = ROOT / weights_name
        model = YOLO(weights if weights.exists() else weights_name)
        _TASK_NAMES_CACHE[task_name] = {int(k): str(v) for k, v in model.names.items()}
    return _TASK_NAMES_CACHE[task_name]


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",

View source record →