
Mobilenetv2: TensorFlow2中的MobileNetV2实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目为TensorFlow 2版本的MobileNetV2模型实现,适用于资源受限的设备上高效运行深度学习应用。
具有完整源代码的TensorFlow2版本的MobileNetv2。
使用Kaggle dogs-vs-cats数据集从头开始训练MobileNetv2模型。当前状态是在ZCU102上测试,使用的工具是带有额外补丁的TensorFlow2.3和Vitis AI 1.3.2。
我们将执行以下步骤:
- 下载并准备Kaggle dogs-vs-cats数据集。
- 将图像转换为TFRecords格式。
- 使用TensorFlow内置的Keras版本对自定义CNN进行训练和评估。
- 使用作为Vitis AI一部分提供的Xilinx量化器,将浮点模型转化为量化模型。
- 用dogs-vs-cats测试数据集来评估这个量化后的模型。
- 编译量化模型以备在目标板上执行。
- 利用提供的Python脚本,在目标板上运行。
全部评论 (0)
还没有任何评论哟~


