cnn_tf_graphs.py 文件源码

def confidence_cnn13(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)

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

    #print ('final conf', final_confidence)

    return final_confidence