INNER CODE UNIT · Python

weights

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

        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",
        "author": "Ultralytics",
        "date": time.strftime("%Y-%m-%d"),
        "version": ultralytics_version,
        "task": task_name,
        "batch": 1,
        "imgsz": [task.imgsz, task.imgsz],

View source record →