MobileNet V2在Coral Dev Board的部署过程

阅读: 评论:0

MobileNet V2在Coral Dev Board的部署过程

MobileNet V2在Coral Dev Board的部署过程

MobileNet V2在Coral Dev Board的部署过程

基于google给出的训练自己的模型的示例

.ipynb#scrollTo=RwywT4ZpQjLf

把一个模型部署到coral的edgeTPU的流程是这样的:

1, 用Tensorflow 2.0建立模型

2, 用某个数据集对模型进行训练,生成对应的labels文件

3, 将模型量化,生成.tflite文件

4, 将量化后的.tflite文件对edgeTPU进行编译,生成带有_edgetpu后缀的.tflite文件

5, 将labels文件和_edgetpu文件上传至coral并运行

一, 用Tensorflow2.0建立模型

由于TFlite是在TF的基础上进行转换,所有模型的建立需要用TF2.0

这次我们用的模型是MobileNet V2,Google官方给出的源代码是用TF1.x写的,于是我在网上找了一个别人通过TF2.0实现的

通过del调用模型

二, 用某个数据集进行训练,生成对应的labels文件

因为这次主要目的是观察模型结构与运算时间的关系,对模型的性能没有要求,

所以就用了google示例文件里的5种花的数据集进行训练

(主要是因为不用改代码不会出错

之前尝试了用cifar10数据集进行训练,只是在量化过程中需要对代码进行额外的调整

为了在测试中进行分类显示,需要自动(或手动)生成一个格式如下的labels文件(.txt)

本文发布于:2024-02-04 21:12:02,感谢您对本站的认可!

本文链接:https://www.4u4v.net/it/170716510059647.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:过程   MobileNet   Coral   Board   Dev
留言与评论(共有 0 条评论)
   
验证码:

Copyright ©2019-2022 Comsenz Inc.Powered by ©

网站地图1 网站地图2 网站地图3 网站地图4 网站地图5 网站地图6 网站地图7 网站地图8 网站地图9 网站地图10 网站地图11 网站地图12 网站地图13 网站地图14 网站地图15 网站地图16 网站地图17 网站地图18 网站地图19 网站地图20 网站地图21 网站地图22/a> 网站地图23