具有完整源代码的TensorFlow2版本的MobileNetv2
具有完整源代码的TensorFlow2版本的MobileNetv2。 使用Kaggle dogs-vs-cats数据集从头开始训练MobileNetv2模型。
当前状态
在ZCU102上测试
使用的工具:带有额外补丁的TensorFlow2.3和Vitis AI 1.3.2
数据集:
网络:MobileNetv2
介绍
我们将执行以下步骤:
下载并准备Kaggle dogs-vs-cats数据集。 将图像转换为TFRecords。
使用TensorFlow的内置Keras版本对自定义CNN进行培训和评估。
使用作为Vitis AI的一部分提供的Xilinx:registered:量化器对浮点模型进行量化。
使用dogs-vs-cats测试数据集评估量化模型。
编译量化模型以在目标板上执行。
使用提供的Python脚本在目标板上
2021-07-06 09:02:15
30KB
Python
1