INNER CODE UNIT · Python

main

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

def main() -> None:
    """Export requested YOLO LiteRT release assets."""
    args = parse_args()
    from ultralytics import __version__
    from ultralytics.utils.checks import check_version

    check_version(__version__, ">=8.4.142", name="ultralytics", hard=True)
    output_dir = args.output_dir.resolve()
    output_dir.mkdir(parents=True, exist_ok=True)
    if args.worker_model_id:
        if args.worker_imgsz is None:
            raise ValueError("--worker-imgsz is required with --worker-model-id")
        export_one(args.worker_model_id, args.worker_imgsz, output_dir)
        return

    os.chdir(output_dir)
    release_dir = output_dir / "release-assets"
    release_dir.mkdir(parents=True, exist_ok=True)

View source record →