Как предсказать другую модель в пользовательской функции потерь - PullRequest
Новая
       6

Как предсказать другую модель в пользовательской функции потерь

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

Я пытаюсь предсказать другую модель в моей пользовательской функции потерь. Модель возьмет изображение размером (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 и использовать их значение для расчета потерь.

...