Tensorflow Lite GPUdelegate выдает ошибку «Размеры не BHWC» для последнего слоя.Но мне кажется, что слой, как и все остальные слои, BHWC - PullRequest
Новая
       12

Tensorflow Lite GPUdelegate выдает ошибку «Размеры не BHWC» для последнего слоя.Но мне кажется, что слой, как и все остальные слои, BHWC

1 голос
/ 16 апреля 2019

Я пытаюсь реализовать детектор YOLO на Android. Я следовал демонстрационному коду, предоставленному tenorflow. Модель отлично работает на процессоре или с 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)

Модель отлично работает на ПК с процессором или графическим процессором.

...