vitis AI使用方法
时间: 2023-05-26 08:02:04 浏览: 91
Vitis AI 是 Xilinx 公司推出的面向边缘AI应用的综合开发环境,可以支持Keras,TensorFlow和Caffe等一系列深度学习框架的模型优化和部署。以下是Vitis AI的使用方法:
1. 下载并安装Vitis AI
Vitis AI可在Xilinx公司简介页下载,也可在官网上下载。下载后按照安装说明进行安装。
2. 创建项目
创建Vitis AI项目需要 Vitis AI 以及支持的硬件平台。平台支持 Ultra96、ZCU104、Alveo U250 等等,开发板需要使用 PYNQ 作为操作系统。在PYNQ环境下,可以打开缺省安装好的终端,进入Vitis AI目录下,使用以下命令创建项目.
```
source /opt/vitis_ai/setup.sh
```
```
vitis_ai_compiler
```
3. 选定深度学习框架
选定需要使用的框架,Vitis AI支持的框架包括TensorFlow、Caffe、Keras和ONNX等。对于Pynq而言,需要 手动编译Caffe或TensorFlow等框架,也可以选择已经编译好的模型。
4. 选择模型
用户需要将模型转化为量化模型或是进行裁剪时,需调用Xilinx提供的quantizer和pruner工具一步步进行操作。如果需要直接使用一个已经训练好的模型,可以选择 Vitis AI 中预训练好的模型。
5. 优化模型
Vitis AI在使用量化和裁剪等功能后,可以使用针对AI加速的DPU IP作为部署目标。使用DPU IP的目的是,DPU IP是专为AI推理加速而设计的硬件单元,并且在 Vitis AI 中已经对完整的神经网络架构做了适配的模型库。
6. 部署和测试
完成模型优化和调试后,使用Vitis AI提供的shell脚本及SDK对模型进行编译和测试。编译后部署到FPGA板子上即可,执行一段测试脚本即可运行并评估程 度。
以上便是Vitis AI的简略使用方法。由于平台及硬件不同,详细的操作流程会略有不同。用户应该根据输入输出数据及框架选择手册中对应的执行方式。
相关推荐

















