import tensorflow as tf


def object_loss(scores, labels):
    with tf.variable_scope('object_loss'):
        loss_vector = tf.nn.softmax_cross_entropy_with_logits(
            scores,
            labels,
            name='softmax_cross_entropy_with_logits')

        loss = tf.reduce_mean(
            loss_vector, 
            name='average_loss')

    return loss


def attribute_loss(scores, labels, batch_size):
    with tf.variable_scope('attribute_loss'):
        loss_matrix = tf.nn.sigmoid_cross_entropy_with_logits(
            scores,
            labels,
            name='sigmoid_cross_entropy_with_logits')

        label_count = tf.reduce_sum(
            labels, 
            0,
            keep_dims=True,
            name='label_count')

        batch_size = tf.constant(batch_size, dtype=tf.float32)

        w1 = ((1.0-labels)*label_count/batch_size)
        w2 = (labels*(batch_size-label_count)/batch_size)
        w = w1 + w2
        loss = tf.reduce_mean(
            w*loss_matrix,
            name='average_loss')

    return loss


def regularization_loss(param_list, coeff):
    regularizer = tf.zeros(shape=[])
    for param in param_list:
        regularizer += tf.nn.l2_loss(param) 
    return coeff*regularizer

if __name__=='__main__':
    scores = tf.constant([[0.2, 0.3, 0.7],[0.8, 0.2, 0.9]])
    labels = tf.constant([[1.0, 0.0, 0.0],[0.0, 1.0, 0.0]])
    loss = attribute_loss(scores, labels)
    sess = tf.InteractiveSession()
    with sess.as_default():
        print loss.eval()