Я пытаюсь предсказать другую модель в моей пользовательской функции потерь. Модель возьмет изображение размером (224,224,3) и выдаст за него эстетику c. Я пытаюсь ввести туда свой y_pred, но пока он дает всевозможные ошибки. Кто-нибудь знает правильный способ предсказать другую модель в пользовательской функции потерь? Код для функции потерь был прикреплен ниже.
def NIMA_Loss(y_true,y_pred):
gamma=0.001
num_ex = K.shape(y_pred)[0]
y_p = K.reshape(y_pred, (num_ex,224,224,3)))
img_tensor=tf.expand_dims(y_p,axis=0)
img_tensor=preprocess_input(img_tensor)
scores=nima_model.predict(img_tensor)
finalScore=mean_score(scores)+std_score(scores)
finalLoss=metrics.mae(y_true,y_pred)+gamma*(10-score)
return finalLoss
Выходное изображение имеет размер (512,512,3), поэтому я изменяю его на (224,223,3). Подводя итог, моя главная цель - скормить мои y_pred до nima_model и использовать их значение для расчета потерь.