search

ry / tensorflow-vgg16

conversation of caffe vgg16 model to tensorflow
672 stars 285 forks source link

How to get the tensor name #15

Open lipanpeng opened 6 years ago

lipanpeng commented 6 years ago

I want to extract features with vgg16. I can use "get_tensor_by_name" to get the features of certain layer. Now I don't the exact name of tensor. So how can I get the all the name of tensors in network?

TetroAlon commented 6 years ago

HI lipanpeng ... just save the model as ".ckpt" file.

Then you'll have easy access to your network layers. look here: https://www.tensorflow.org/programmers_guide/saved_model