Tensorflow, обслуживающий операции contrib

Question

Как мне обслуживать модель с тензорным потоком , если есть операции tf.contrib.Я использую Tensorflow Serving via Docker (последняя версия) (версия tf 1.11), и когда я обслуживаю модель, появляется следующее сообщение:

“Failed to start server. Error: Unknown: 1 servable(s) did not become available: {{{name: slider_universal version: 1} due to error: Not found: Op type not registered ‘ImageProjectiveTransformV2’ in binary running on 254345a5d9f1. Make sure the Op and Kernel are registered in the binary running in this process. Note that if you are loading a saved graph which used ops from tf.contrib, accessing (e.g.) tf.contrib.resampler should be done before importing the graph, as contrib ops are lazily registered when the module is first accessed.}, }”

Я также построил с помощью Bazel, но была та же ошибка

Я использую tf.contrib.image.transform

Если я удаляю эту операцию во время экспорта модели, она может обслуживаться тензорным потоком, обслуживающим

Answer 1

У меня были проблемы с той же операцией, и кажется, что единственный способ - это построить ее.

tensorflow_serving/model_servers/BUILD определяет, какие операции Tensorflow будут включены в переменную SUPPORTED_TENSORFLOW_OPS, и я был немного сбит с толкуоб этом, так как он указывает, что обязательные операции должны быть включены.Тем не менее, поскольку в правиле сборки contrib-тензорного потока, по-видимому, нет опций в contrib.image, я вместо этого явно добавил их, обновив эту переменную до следующего

SUPPORTED_TENSORFLOW_OPS = [
    "@org_tensorflow//tensorflow/contrib:contrib_kernels",
    "@org_tensorflow//tensorflow/contrib:contrib_ops_op_lib",
    "@org_tensorflow//tensorflow/contrib/image:image_ops_kernels",
    "@org_tensorflow//tensorflow/contrib/image:image_ops_op_lib",
]