时间戳

首页 > 解决方案 > Tensorflow Lite GPUdelegate 在最后一层给出“Dimensions are not BHWC”错误。但对我来说，这一层似乎是 BHWC，就像其他所有层一样

android - Tensorflow Lite GPUdelegate 在最后一层给出“Dimensions are not BHWC”错误。但对我来说，这一层似乎是 BHWC，就像其他所有层一样

问题描述

我正在尝试在 android 上实现一个 YOLO 检测器。我按照 tensorflow 提供的演示代码进行操作。该模型在 CPU 或 NNAPI 上运行良好。但是，当我尝试使用 GPU 运行它时，程序崩溃并且调试器给出以下错误：

    java.lang.IllegalArgumentException: Internal error: Failed to apply delegate: GpuDelegate Prepare: Dimensions are not BHWCNode number 23 (GpuDelegate) failed to prepare.

模型的最后一层看起来像self.conv9 = layers.Conv2D(425, (1,1), strides=(1,1), padding='same', name='conv_9', use_bias=False)

该模型在带有 CPU 或 GPU 的 PC 上运行良好。

标签： androidtensorflowgputensorflow-lite

解决方案

推荐阅读