def confidence_cnn14(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_cnn4(image, input_size)
final_confidence = tf.where(visable, confidence, tf.zeros_like(confidence))
#print ('final conf', final_confidence)
return final_confidence
评论列表
文章目录