Tensorflow从重新训练的模型中提取pool3特征的速度很慢

时间:2016-06-21 12:36:24

标签: tensorflow

我使用重新训练代码在我自己的图像上训练初始模型的最后一层。但是在我的CPU机器上,提取pool3功能几乎需要0.5-0.8秒。

pool3 = sess.graph.get_tensor_by_name('pool_3:0')
pf = sess.run(pool3, {png_data: imgBuffer.getvalue()})

上面的代码行总是占用。是因为我的CPU内存吗?有什么方法可以让它更快?

1 个答案:

答案 0 :(得分:0)

Tensorflow使用惰性评估;基本上所有图中的节点都是根据需要在sess.run()进行评估。

这就是为什么你注意到sess.run花费了大部分时间,而不是特别是pool3更昂贵

您可以使用Timeline查找详细的时间细分。