步骤 2:安装 AI 模型程序包 - 2.5 简体中文

Vitis AI Library 用户指南 (UG1354)

Document ID
UG1354
Release Date
2022-06-15
Version
2.5 简体中文

Vitis AI 运行时包和 Vitis AI Library 样本与模型都已内置在开发板镜像中。因此,您无需在开发板上单独安装 Vitis AI 运行时包和模型包。但是,您仍可在自己的镜像上或者在官方镜像上遵循如下步骤安装模型:

  1. 对于每个模型,都有一个 yaml 文件用于描述有关模型的详细信息。此 yaml 文件包含适用于不同平台的下载链接。选择您的模型和所需平台并下载。
  2. 使用 scp 搭配以下命令将下载的文件复制到开发板。
    $ scp <model>.tar.gz root@IP_OF_BOARD:~/

    如果您的开发板连接到互联网,您也可以使用 wget 命令将模型直接下载到开发板。

  3. 登录到开发板(使用 ssh 或串行端口)并安装模型程序包。
  4. 如果 /usr/share/vitis_ai_library/model 文件夹不存在,请首先创建此文件夹。
    # mkdir -p /usr/share/vitis_ai_library/models
  5. 在目标侧安装模型。
    # tar -xzvf <model>.tar.gz -C /usr/share/vitis_ai_library/models

    默认情况下,这些模型位于目标侧的 /usr/share/vitis_ai_library/models 目录中。