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
最后
以上就是等待发箍最近收集整理的关于Vitis-AI-1.3-TensorFlow2 使用TensorFlow2 构建custom AI模型的流程;的全部内容,更多相关Vitis-AI-1.3-TensorFlow2内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复