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 頁面獲得源碼

#訓練模型
[[email protected] src]$ python3.8 train.py

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