ValueError: Неизвестный слой: «CustomScaleLayer» ⇐ Python

[Anonymous]

Я создаю модель с помощью InceptionResNETV2 для обучения набору данных изображений.
Моя модель выглядит так:

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

`inception_resnet = InceptionResNetV2(weights = "imagenet",
include_top = False,
input_tensor = Input(shape = (224, 224, 3)))
inception_resnet.trainable = False

#--------------------------------------------------------------------------------------------
headmodel = inception_resnet.output
headmodel = Flatten()(headmodel)
headmodel = Dense(500, activation = "relu")(headmodel)
headmodel = Dense(250, activation = "relu")(headmodel)
headmodel = Dense(4, activation = "sigmoid")(headmodel)

#---------------------------------------------------------------------------------------------

model = Model(inputs = inception_resnet.input, outputs = headmodel)
model.compile(loss = 'mse', optimizer = tf.keras.optimizers.Adam(learning_rate = 1e-4))
model.summary()

Я обучил свой набор данных на следующей модели и сохранил его как
model.save('detection.h5')
Когда я пытаюсь загрузить сохраненную модель для прогнозирования, используя
model = tf.keras.models.load_model('detection.h5')
Он выдает ошибку:
ValueError: Unknown Layer: 'CustomScaleLayer'.
Может ли кто-нибудь помочь мне с загрузкой модели для прогнозирования...
Я пробовал использовать
model = tf.keras.models.load_model((path),custom_objects={'KerasLayer':hub.KerasLayer})
Но это не решило мою проблему.