适用于嵌入式系统ARM6的 tf.keras.layers.experimental.preprocessing.Resizing 的替代方案

问题描述

我使用以下模型进行音频分类

model = models.Sequential([
  layers.Input(shape=input_shape),preprocessing.Resizing(64,64),layers.Conv2D(64,3,activation='relu',padding="same"),activation='relu'),layers.MaxPooling2D(),layers.Conv2D(128,layers.Conv2D(32,layers.Dropout(0.25),layers.Flatten(),layers.Dense(128,layers.Dropout(0.5),layers.Dense(2),])

这在 cpu 上运行良好。现在我想在 TPU(通过 RaspBerry Pi Zero (ARM6) 的 Coral USB Accelerator)上运行经过训练的模型。在 Coral 上运行模型必须使用的 edge TPU compiler 不接受 tf.keras.layers.experimental.preprocessing.Resizing代码片段中的第三行)。边缘 TPU 编译器只接受 TensorFlow lite 函数一个子集。因此,我现在正在寻找一种替代方法,将我的 624x129x1 频谱图调整为 64x64x1,作为models.Sequential() 之外的预处理步骤。不幸的是,Tensorflow 和 PyTorch 不适用于 RPI0。因此,我无法使用这些库,并且正在寻找将频谱图预处理为正确大小的替代方法。 非常感谢任何帮助或建议!

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)