Merge pull request opencv#595 from dkurt:tf_squeeze_and_slim_softmax_v2

Author: alalek

testdata/dnn/tensorflow/generate_tf_models.py

@@ -703,6 +703,27 @@ def pad_depth(x, desired_channels):
         f.write(graph_def.SerializeToString())
 
 ################################################################################
+# issue https://github.com/opencv/opencv/issues/13839
+inp_node = 'PNet/conv3/add'
+out_node = 'PNet/cls_prob'
+with tf.Session(graph=tf.Graph()) as localSession:
+    localSession.graph.as_default()
+
+    with tf.gfile.FastGFile('PNet_pnet.pb') as f:
+        graph_def = tf.GraphDef()
+        graph_def.ParseFromString(f.read())
+        graph_def = TransformGraph(graph_def, [inp_node], [out_node], ['strip_unused_nodes'])
+
+    tf.import_graph_def(graph_def, name='')
+
+    inputData = gen_data(tf.placeholder(tf.float32, [1, 4, 5, 16], inp_node))
+    outputData = localSession.run(localSession.graph.get_tensor_by_name(out_node + ':0'),
+                                  feed_dict={inp_node + ':0': inputData})
+    writeBlob(inputData, 'slim_softmax_v2_in')
+    writeBlob(outputData, 'slim_softmax_v2_out')
+
+    with tf.gfile.FastGFile('slim_softmax_v2_net.pb', 'wb') as f:
+        f.write(graph_def.SerializeToString())
 
 # Uncomment to print the final graph.
 # with tf.gfile.FastGFile('fused_batch_norm_net.pb') as f: