TensorFlow的pb文件相关操作
import tensorflow as tf
gf = tf.GraphDef()
gf.ParseFromString(open('model_action_recog_tiny_8_LSTM_51213.pb','rb').read())
a=[n.name + '=>' + n.op for n in gf.node if n.op in ( 'Softmax','Placehol...
原创
2019-11-11 16:22:21 ·
165 阅读 ·
0 评论