Vitis-AI-1.3-TensorFlow2 使用TensorFlow2 构建custom AI模型的流程;

Vitis-AI 1.3/TensorFlow2 环境下使用自定义 CNN 模型进行 Mnist 分类
本示例工程中我们会在 TensorFlow2 下使用 Keras API 创建一个自定义 CNN 网络,在 Vitis-AI 1.3 环境下编译成 Xilinx DPU 上运行的模型文件,并在 Xilinx zynqMP 上部署运行。

可从link的 github 页面获得源码

TensorFlow model -> Quantizer -> Compiler -> xmodel (deployable model in Xilinx DPU)

Vitis-AI-1.3-TensorFlow2 使用TensorFlow2 构建custom AI模型的流程;_第1张图片

Python scripts

load_data.py: load Mnist dataset

  • generate_images.py: Generate local images from Keras datasets. This file is form https://github.com/Xilinx/Vitis-AI-Tutorials/tree/MNIST-Classification-TensorFlow

你可能感兴趣的:(FPGA,opencl,opencv,vitis,tensorflow)