Tensorflow：將預測導出到數組或文件

Question

由於所有計算都是在會話下進行的，有沒有辦法將Tensorflow的預測導出爲Numpy/Pandas數組或文件，即CSV或TXT？

謝謝！ 保羅

Answer 1

你想使用類似...

im = Image.open('/home/kendall/Desktop/HA900_frames/frame0635.tif') 
batch_x = np.array(im) 
batch_x = batch_x.reshape((1, n_steps, n_input)) 
batch_x = batch_x.astype(float) 
prediction = sess.run(pred, feed_dict={x: batch_x}) 
prediction = np.asarray(prediction) 
prediction = np.reshape(prediction, (200,200)) 
np.savetxt("prediction.csv", prediction, delimiter=",") 

Answer 2

您可以使用TFRecordWriter將輸出寫入一些文件。 https://www.tensorflow.org/api_docs/python/tf/python_io/TFRecordWriter

下面的文章可以給你如何做到這一點的想法： http://warmspringwinds.github.io/tensorflow/tf-slim/2016/12/21/tfrecords-guide/ http://machinelearninguru.com/deep_learning/tensorflow/basics/tfrecord/tfrecord.html