1

我已经尝试了 2 年(不是连续的,转移到其他事情然后回去)在视觉工具包中编译和运行基于 keras.applications 的模型,我尝试了很多方法都没有成功(甚至忘记了一些),问了一些问题在论坛中,在官方项目 repo 中,堆栈溢出等没有运气,例如:

在问了许多不同的问题后,我想可能直接发布我的用例并直接询问可能更有希望:

我需要编译并运行基于 keras.applications 的模型才能在 aiy 视觉套件中运行它,我知道设备是有限的,所以我正在尝试使用文档说视觉套件支持的模型:MobilenetV2 和我'正在通过冻结一些层并从 keras.applications.MobileNetV2 中删除其他层进行迁移学习,然后添加自定义可训练层,以进行测试,我也在尝试 VGG16,过去有很多问题,但最近的问题是:

  1. 即使只使用预训练模型的第一层并丢弃其余部分(导出的 .pb 文件很小,大约 2.5 mb)我得到(对于 vgg16,这发生在我的计算机上编译时,而不是在树莓设备中) : 没有足够的设备内存来运行模型。

  2. 对于 MobileNetv2,即使文档说它受支持,我也会得到:检查失败:other_op->type == OperatorType::kTensorFlowMerge

对我的案子有什么建议吗?还是根本不可能在视觉套件中运行基于 keras.applications 的模型?如果不可能,是否可以将诗人 mobilenet .pb 文件的 tf 与 keras 输出层结合起来并编译那个?

真的很感激对此的一些帮助,或者至少是一个明确的帮助:“不,这是不可能的”,所以我不会继续追求不可能的事情。

4

0 回答 0