且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

Tensorflow:如何从预先训练的CNN的特定层提取图像特征?

更新时间:2022-05-04 00:45:27

除了Peter Hawkins提到的方法外,在注释中使用tf.Graph.get_operations()的一种方法是使用tensorboard查找名称.您想从中提取的图层.

One approach other than what was mentioned by Peter Hawkins, to use tf.Graph.get_operations() in the comments is to use tensorboard to find the name of the layer you would like to extract from.

从那里您可以使用

graph.get_tensor_by_name("import/layer_name")

提取所需的任何功能.