Проблема с загрузкой весов .h5 в модель modellib.MaskRCNN. ⇐ Python

[Anonymous]

Я скачал веса Mask_rcnn_coco.h5 с https://github.com/matterport/Mask_RCNN. Я хочу провести тренировку на предварительно обученной модели с использованием этих весов. Но когда я пытаюсь рассчитать веса, используя следующий код:

Код: Выделить всё

COCO_WEIGHTS_PATH = os.path.join(ROOT_DIR, "mask_rcnn_coco.h5")

config = CustomConfig()
model = modellib.MaskRCNN(mode="training", config=config,
model_dir=DEFAULT_LOGS_DIR)

weights_path = COCO_WEIGHTS_PATH
# Download weights file

if not os.path.exists(weights_path):
utils.download_trained_weights(weights_path)s

model.load_weights(weights_path, by_name=True, exclude=[
"mrcnn_class_logits", "mrcnn_bbox_fc",
"mrcnn_bbox", "mrcnn_mask"])

train(model)

Я получаю сообщение об ошибке NotImplementedError: сохранить или восстановить веса, которые не являются экземпляром tf. Переменная не поддерживается в h5, вместо этого используйте save_format='tf'. Получили модель или слой Conv2D с весами [, ]
Я пытался загрузить веса, а затем сохранить их, но ошибка продолжает возникать. Есть ли способ решить эту проблему?

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