python - TensorFlow Lite for Microcontrollers 中的“支持的操作”是什么意思？

Question

我想使用 OpenMV Cam H7 和 tensorflow 创建用于面部识别的图像分类模型。在tensorflow 文档中解释说“用于微控制器的 TensorFlow Lite 目前支持有限的 TensorFlow 操作子集，这会影响可以运行的模型架构”

“支持的操作可以在文件all_ops_resolver.cc中看到”

那么什么是支持的操作？，以及我如何知道我在模型中使用了哪些支持的操作

Answer 1

如果您打开刚刚共享的 all_ops_resolver.cc 文件的链接，您将能够看到支持的操作列表。该列表包括设计面部识别模型所需的基本构建块，例如 Conv2D/DepthwiseConv2D 和 FullyConnected 层，以及 Relu 和 Softmax 等激活。

要查看模型中使用的层，您只需在 Tensorflow/Keras 中调用 model.summary()。

我建议您从寻找一个简单的 Tensorflow 示例开始，该示例展示如何构建人脸识别模型，并尝试仅使用 Tensorflow-lite for Microcontrollers 支持的操作来构建类似的模型。