Ошибка типа: load_model () отсутствует 1 обязательный позиционный аргумент: 'filepath' - PullRequest
0 голосов
/ 03 марта 2020

Я использую Keras 2.2.5, и на этапе обучения я сохранил модель с функцией ModelCheckpoint, которая импортируется следующим образом: from keras.callbacks import ModelCheckpoint

Затем, на этапе тестирования, когда я хочу чтобы загрузить модель с помощью функции load_model (from keras.models import load_model), я получаю ошибку заголовка.

Тестовый скрипт выглядит следующим образом:

import numpy as np
import argparse
import keras.layers as KL
from keras.models import load_model
from keras.preprocessing import image
from keras.applications.resnet50 import preprocess_input as preprocess_input_resnet
if __name__ == "__main__":
    # Define variables
    parser = argparse.ArgumentParser()
    parser.add_argument("--image_size", type=tuple, default=(500, 500))
    parser.add_argument("--mask_size", type=tuple, default=(32,32))
    parser.add_argument("--image_path", type=str, default="../DATA/resized_imgs/13056.png")
    parser.add_argument("--mask_path", type=str, default="../DATA/resized_masks/13056.png")
    parser.add_argument("--path_of_the_checkpoint", type=str, default="./RESULTS/2020_02_25_12_34_54/bestmodel/MultiLabel_PETA_weights.best.hdf5")
    parser.add_argument("--Categories", type=list, default=["personalLess30", "personalLess45", "personalLess60", "personalLarger60"])
    args = parser.parse_args()

    # Load trained model
    PAR_model = load_model (filepath=args.path_of_the_checkpoint, custom_objects={'BatchNorm':KL.BatchNormalization})
    PAR_model.summary()

1 Ответ

0 голосов
/ 03 марта 2020

Вы не должны передавать позиционные аргументы в качестве аргументов ключевого слова:

PAR_model = load_model (args.path_of_the_checkpoint, custom_objects={'BatchNorm':KL.BatchNormalization})
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...