Маска R-CNN: Как определить форму выходных данных Lambda? ⇐ Python

[Anonymous]

Я изучаю следующий пример кода Mask R-CNN:
https://github.com/matterport/Mask_RCNN ... er/samples
Однако когда я запустил файл demo.ipynb, для приведенного ниже кода была возвращена ошибка, которую я не знаю, как исправить.
model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
Ошибка выглядит следующим образом.
NotImplementedError: Exception encountered when calling Lambda.call().
We could not automatically infer the shape of the Lambda's output. Please specify the `output_shape` argument for this Lambda layer.
Arguments received by Lambda.call():
• args=('\',)
• kwargs={'mask': 'None'}

Похоже, что функция fpn_classifier_graph вызвала ошибку, поскольку не смогла идентифицировать выходные данные Lambda, и необходимо указать правильную выходную_форму.
Однако я понятия не имею, как изменить код...
ChatGPT предложил пересмотреть Lambda, чтобы явно указать выходную_форму следующим образом, что не сработало... р>
shared = KL.Lambda(lambda x: K.squeeze(K.squeeze(x, 3), 2),
name="pool_squeeze",
output_shape=lambda s: (s[0], s[1], s[3]))(x)


Подробнее здесь: https://stackoverflow.com/questions/785 ... das-output