如何为coral培训高效的edgeTPU模型?

2024-05-29 08:11:06 发布

您现在位置:Python中文网/ 问答频道 /正文

我有珊瑚USB加速器,我正在搜索训练图像分类模型。我能找到的最适合我需要的型号是EfficientNet EdgeTpu。我正试图找到一种方法,以tensoflow API为例,在我的计算机中重新训练模型,但我能找到的(主要是coral page和tensorflow git readme)就是如何使用谷歌云工具和虚拟机对其进行训练。是否有必要使用谷歌云解决方案来培训高效的网络EdgeTpu模型

我还发现了如何轻松训练Efficientnet型号B0-B7,但我不知道该型号是否与在coral USB中运行的Efficientnet EdgeTpu(S/M/B)相同

我只是想找到一个简单的方法,在我的电脑中训练高效的网络模型,它更适合coral USB,而不必创建一个帐户并支付谷歌存储和虚拟机培训的费用


Tags: 方法模型图像网络分类usb型号加速器
1条回答
网友
1楼 · 发布于 2024-05-29 08:11:06

您可以做的一件事是在自定义数据上训练开箱即用的efficientnet(来自tensorflow),并将其转换为tflite,然后使用edgetpu_编译器对其进行编译

from tensorflow.keras.applications import EfficientNetB0

相关问题 更多 >

    热门问题