提交 ba98bf87 编写于 作者: 李滨

Merge branch 'master' into 'master'

Add softmax

See merge request !1168
......@@ -37,6 +37,7 @@ def tensors_to_images(input_files, image_shape):
output = tf.placeholder(tf.string, name='output_file')
# use the second channel if it is gray image
if image_shape[2] == 2:
input = tf.nn.softmax(input)
_, input = tf.split(input, 2, axis=2)
tensor_data = tf.image.convert_image_dtype(input,
tf.uint8,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册