Когда я попытался развернуть модель MASKRCNN https://github.com/matterport/Mask_RCNN на платформе Google AI, я получил эту ошибку:
WARNING: Using endpoint [https://ml.googleapis.com/]
Creating version (this might take a few minutes)......failed.
ERROR: (gcloud.ai-platform.versions.create) Create Version failed. Model validation failed: Outer dimension for outputs must be unknown, outer dimension of 'mrcnn_detection/Reshape_1:0' is 1
Любая идея, как ее решить, вот подпись def моя модель:
signature_def['serving_default']:
The given SavedModel SignatureDef contains the following input(s):
inputs['input_anchors'] tensor_info:
dtype: DT_FLOAT
shape: (-1, -1, 4)
name: input_anchors:0
inputs['input_image'] tensor_info:
dtype: DT_FLOAT
shape: (-1, -1, -1, 3)
name: input_image:0
inputs['input_image_meta'] tensor_info:
dtype: DT_FLOAT
shape: (-1, 14)
name: input_image_meta:0
The given SavedModel SignatureDef contains the following output(s):
outputs['mrcnn_detection/Reshape_1'] tensor_info:
dtype: DT_FLOAT
shape: (1, 100, 6)
name: mrcnn_detection/Reshape_1:0
outputs['mrcnn_mask/Reshape_1'] tensor_info:
dtype: DT_FLOAT
shape: (1, 100, 28, 28, 2)
name: mrcnn_mask/Reshape_1:0
Method name is: tensorflow/serving/predict
Вот сценарий, который я использую для создания модели tf-обслуживания: https://github.com/moganesyan/tensorflow_model_deployment/blob/master/tf_serving/make_tf_serving.py
Как можно исправить эту ошибку?
Я сослался на эту проблему , но не смог найти решение solid.