我想使用从一组图像中提取的pool_3功能。目前,我在每个图像上都有一个循环以提取pool_3功能:
# X_input.shape = (40000, 32, 32, 3)
def batch_pool3_features(X_input):
sess = tf.InteractiveSession()
n_train = X_input.shape[0]
print 'Extracting features for %i rows' % n_train
pool3 = sess.graph.get_tensor_by_name('pool_3:0')
X_pool3 = []
for i in range(n_train):
print 'Iteration %i' % i
pool3_features = sess.run(pool3,{'DecodeJpeg:0': X_input[i,:]})
X_pool3.append(np.squeeze(pool3_features))
return np.array(X_pool3)
不过这很慢。有更快的批处理实现吗?
谢谢
最佳答案
尚未-尚未。我已经打开a ticket for this feature request on github来回答另一个问题。