Ошибка преобразования Tensorflowjs: «Ошибка ValueE: неподдерживаемые операции» - PullRequest
0 голосов
/ 27 февраля 2019

Я конвертировал свою tensorflow модель в tensorflowjs форму, используя следующую команду

tensorflowjs_converter 
    --input_format=tf_saved_model 
    --output_node_names="my_output_node" \
    --saved_model_tags=serve my_saved_model_dir \
    ./web_model

Я вызвал следующую загадочную ошибку:

ValueError: Unsupported Ops in the model before optimization NonMaxSuppression, ResizeArea

У меня есть эти операции в моем графике.Нужно ли мне поменять их на что-то более tensorflowjs дружественное?

Я углубился в Google и наткнулся только на ссылку на следующий флаг, который я мог добавить к команде tensorflowjs _converter --skip_op_check=SKIP_OP_CHECK \ Thisдействительно скомпилировался, но затем при попытке обслужить модель js я обнаружил ошибку js, похожую на приведенную выше:

Ошибка: Tensorflow Op не поддерживается: ResizeArea

Есть идеи, как изменить мой график или мою команду для навигации?

Спасибо

1 Ответ

0 голосов
/ 27 февраля 2019

Краткий ответ - да, вам нужно их изменить.TensorflowJS изменит операции в целях оптимизации, но не все операции имеют эквивалентную версию TFJS.Полный список поддерживаемых операций приведен здесь: https://github.com/tensorflow/tfjs-converter/blob/master/docs/supported_ops.md

Как ни странно, «NonMaxSuppression» действительно присутствует в списке, но ResizeArea нет, и 100% не будет работать.

Альтернативаэто создать собственную операцию и использовать этот код, но я не уверен, как это сделать в TFJS.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...