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

353 阅读1分钟

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

可从link的 github 页面获得源码

#训练模型
[exxxx@fpga src]$ python3.8 train.py

Load Mnist dataset..
Downloading data from https://storage.googleapis