cnn_tf_graphs.py 文件源码

def confidence_cnn23(image_with_alpha, input_size=512):
    image = tf.slice(image_with_alpha,[0,0,0,0],[-1,-1,-1,3])
    alpha = tf.slice(image_with_alpha,[0,0,0,3],[-1,-1,-1,1])

    #print ('image', image)
    #print ('alpha', alpha)

    visable = tf.not_equal(alpha, tf.zeros_like(alpha))

    confidence = confidence_cnn3(image, input_size)

    negative_confidence = tf.multiply(tf.ones_like(confidence),tf.constant(-1.0))

    final_confidence = tf.where(visable, confidence, negative_confidence)

    #print ('final conf', final_confidence)

    return final_confidence