有没有办法在 RPI 零上使用已编译的 keras 模型?

Is there a way to use a compiled keras model on the RPI Zero?

我正在为机器人开发字母识别应用程序。我使用家用 PC 来训练模型,并希望使用已训练的模型在 RPI 零 W 上进行识别。

我有一个 HDF 模型。当我尝试在 RPI 零上安装 Tensorflow 时,它抛出一个哈希错误,据我所知,这是由于 64 位机器的 TF beeing。当我尝试安装 Tensorflow Lite 时,安装卡顿并崩溃。

为了保存我使用的模型:

classifier.save('test2.h5')

这是预测线:

test_image = ks.preprocessing.image.load_img('image.jpg')
test_image = ks.preprocessing.image.img_to_array(test_image)

result = classifier.predict(test_image)

我也试过通过 Nuitka 编译 python 脚本,但是由于 RPI 是 ARM 并且 nuitka 不提供交叉编译,所以我觉得有这种可能性。

您可以使用已有的 TFLite 来解决您的问题。

如果这没有帮助,您还可以从源代码构建 TFLite。

请参考以下链接:

https://www.tensorflow.org/lite/guide/build_rpi

https://medium.com/@haraldfernengel/compiling-tensorflow-lite-for-a-raspberry-pi-786b1b98e646