Feature extract using VGG

tomrunia / TF_FeatureExtraction

Convenient wrapper for TensorFlow feature extraction from pre-trained models using tf.contrib.slim

180 stars 61 forks source link

Feature extract using VGG #9

Open midmelody opened 6 years ago

midmelody commented 6 years ago

Inception and Resnet are both fine, but when I tried to use vgg_16/19, I came up with some strange error "Invalid argument: Assign requires shapes of both tensors to match", and print_network_summary() did not work. Could you give an example for VGG? Thanks!

zengsn commented 6 years ago

--layer_names=vgg_16/fc8

print_network_summary() works.