loss = tf.losses.mean_squared_error(
    predictions=heatmaps, labels=labels_tensor
)

loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=heatmaps, labels=labels_tensor), name= 'cross_entropy_loss')