如何在自定义损失函数中预测另一个模型

Question

我正在尝试在自定义损失函数中预测另一个模型。该模型将采用大小(224,224,3)的图像，并为其给出美学分数。我正在尝试输入我的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并使用其值来计算损失。