Как я могу исправить эту ошибку от тестирования моей модели rcnn маски - PullRequest
0 голосов
/ 13 октября 2019

enter image description here

Пожалуйста, мне нужно решение с исправлением и запуском этого кода для маски rcnn

1 Ответ

0 голосов
/ 13 октября 2019

Как прокомментировал Gtomika, предоставьте, пожалуйста, трассировку ошибок в текстовом формате. Также предоставьте более подробную информацию о репо, из которого вы получили модель, и любую другую информацию, которая, по вашему мнению, является актуальной.

Исходя из моего прошлого опыта, я почти уверен, что вы используете маску Matterplot RCNN и вашу проблему. из-за несоответствия количества классов. Вы пытаетесь загрузить вес модели, которая была обучена по другому количеству классов. Вы должны исключить некоторые слои, такие как 'mrcnn_bbox_fc', 'mrcnn_class_logits'.

Исправление: изменить model.load_weights(COCO_MODEL_PATH, by_name=True) на model.load_weights(filepath, by_name=True, exclude=[ "mrcnn_class_logits", "mrcnn_bbox_fc", "mrcnn_bbox", "mrcnn_mask"])

Для получения дополнительной информации см. эту проблему .

...