import tensorflow as tf

                                 num_or_size_splits=num_of_joints,
                                 axis=-1)
    losses = []  # 计算每一个关键点的损失值，并累加求平均
    for i in range(num_of_joints):
        heatmap_pred = tf.squeeze(heatmap_pred_list[i])
        heatmap_true = tf.squeeze(heatmap_true_list[i])
        loss = 0.5 * tf.losses.mean_squared_error(y_pred=heatmap_pred * true_weight[:, i],
                                                  y_true=heatmap_true * true_weight[:, i])
