我们可以传递与训练模型不同格式的数据集吗?

时间:2020-01-29 17:07:50

标签: python tensorflow keras tensorflow2.0 tf.keras

我在条目格式为(x, (y_1, y_2, y_3))

的数据集上训练模型

为了进行评估,我会进行后处理:y_pred = f(y_1_pred, y_2_pred, y_3_pred),我想在(x, y)格式的数据集上评估结果。

我可以这样做:

model.fit(x=x_train, y=(y_1_train, y_2_train, y_3_train), validation_data=(x_val, y_val))

假设我的模型已使用自定义指标进行编译,并进行了后处理以检索单个y

就我而言(我正在做人体姿势估计)

  • x_train是图片
  • y_1_train是热图
  • y_2_train是图像中的向量字段
  • y_3_train是图像中的向量字段

从这些我可以重建y_pred,这是每个实例的关键点列表。

x_val是一个实例,而y_val是每个实例的关键点列表。

0 个答案:

没有答案
相关问题