Skip to content

Commit 0e7d1ff

Browse files
committed
Fix Mobilenet v2 from TensorFlow slim
1 parent fd47a5c commit 0e7d1ff

File tree

4 files changed

+5
-0
lines changed

4 files changed

+5
-0
lines changed

testdata/dnn/tensorflow/generate_tf_models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -672,6 +672,11 @@ def pad_depth(x, desired_channels):
672672
sub = conv - inp
673673
save(inp, sub, 'eltwise_sub')
674674
################################################################################
675+
inp = tf.placeholder(tf.float32, [None, 2, 3, 4], 'input')
676+
conv = tf.layers.conv2d(inp, filters=3, kernel_size=[1, 1])
677+
softmax = tf.contrib.slim.softmax(conv)
678+
save(inp, softmax, 'slim_softmax')
679+
################################################################################
675680

676681
# Uncomment to print the final graph.
677682
# with tf.gfile.FastGFile('fused_batch_norm_net.pb') as f:
224 Bytes
Binary file not shown.
854 Bytes
Binary file not shown.
200 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)