Как запустить модель Conv3D в Android? - PullRequest
Новая
       18

Как запустить модель Conv3D в Android?

1 голос
/ 05 мая 2019

Я недавно углубился в мир компьютерного зрения и глубокого обучения. Я разработал 3D-модель CNN для распознавания действий в Керасе, и теперь мне интересно запустить ее в Android (Java). Я использую слои Conv3D и MaxPool3D. Общий размер модели 40MB

Я искал решения в пространстве tenorflow-lite, но похоже, что в них еще не реализованы операции.

Я получил следующую ошибку при использовании функции converter.convert () для получения модели tflite 

ConverterError: TOCO failed. See console for info.
2019-05-05 14:39:07.006669: I tensorflow/lite/toco/import_tensorflow.cc:1336] Converting unsupported operation: Conv3D

Так, что я могу сделать, чтобы иметь возможность запустить его на Java? Должен ли я:

  • запустить файл .pb напрямую? Я даже не знаю, возможно ли это сейчас (после tflite). Если да, сколько времени потребуется смартфону нового поколения для запуска файла размером 40 МБ?
  • Реализовать ops самостоятельно? Если да, то как?
  • попробуйте другой подход вне тензорного потока?
  • реализовать новую архитектуру распознавания действий, которая использует только операции, поддерживаемые tflite
  • прочее

До сих пор я не нашел в Android никакой реализации Conv3D в Android ... Огромное спасибо за внимание!

1 Ответ

0 голосов
/ 06 мая 2019

  • Если вы хотите выполнить в качестве автономного кода JAVA с использованием тензорного потока, посмотрите this .Но если вы хотите реализовать что-то для Android с использованием JAVA, единственный способ - использовать Tensorflow Lite .

  • . Для сравнения, вы можете сравнить свою модель с некоторыми изсовременная архитектура в тестах производительности.Вы можете найти здесь эталонных значений, это показывает сравнение с Pixel 2 и устройством Pixel XL.

  • Для вашей реализации Conv3D, если вы хотите реализовать ops,вы можете посмотреть пользовательских операторов .

  • Я бы предпочел ваше предложение «реализовать новую архитектуру распознавания действий, которая использует только операции, поддерживаемые tflite». Здесь вы можете найти список поддерживаемых операций, используя TF Lite.

...