import tensorflow as tf

        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])
        losses.append(loss)
    return tf.reduce_mean(loss)


class JointsMSELoss(object):
    def __init__(self):
