Блог

Главная - Вопросы по программированию - Маскируйте ошибку формы модели Rcnn при загрузке модели для прогнозирования в потоковом освещении

Маскируйте ошибку формы модели Rcnn при загрузке модели для прогнозирования в потоковом освещении

#python #tensorflow #keras #faster-rcnn

Вопрос:

 ValueError: Layer #391 (named "mrcnn_bbox_fc"), weight <tf.Variable 'mrcnn_bbox_fc_1/kernel:0' shape=(1024, 324) dtype=float32> has shape (1024, 324), but the saved weight has shape (1024, 24).

Я обучил модель MASK-RCNN на пользовательском наборе данных с 5 классами. Когда я выполняю вывод модели в записной книжке и загружаю веса модели с помощью MASK-RCNN, modelib это работает нормально.

 model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)

model.load_weights(weights_path, by_name=True)

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

 import pixellib
from pixellib.instance import instance_segmentation
import cv2
import os

def do_instance(img_path):
    absolute_path = os.path.abspath("model/mask_rcnn_coco.h5")
    segment_image = instance_segmentation(infer_speed="average")
    segment_image.load_model(absolute_path)
    # Perform instance segmentation
    segmask, output = segment_image.segmentImage(img_path, show_bboxes=True)
    return output
Метки: