蟲匯聚之所

環境:Windows底下一堆問題所以被逼無奈安裝了Ubuntu雙系統
使用Anaconda裝了一下tensorflow-gpu版本
沒試過cpu版本不知道跑起來會有什麼差異
參考來源: https://github.com/widiot/tensorflow-practices/tree/master/im2txt 

需要先準備的.
bazel
NumPy 
NTLKtensorflow(使用GPU版本的話還需要安裝CUDA跟CUDNN)

比較特別的是google給的model是tf1.0版本，網路上找到的model幾乎都是tf1.2以上的

所以在使用的時候需要經過轉換才可以順利執行

# 由于版本不同，需要进行修改
def RenameCkpt():
# 1.0.1 : 1.2.1
    vars_to_rename = {
    "lstm/basic_lstm_cell/weights": "lstm/basic_lstm_cell/kernel",
    "lstm/basic_lstm_cell/biases": "lstm/basic_lstm_cell/bias",
    }
    new_checkpoint_vars = {}
    reader = tf.train.NewCheckpointReader(FLAGS.checkpoint_path)
    for old_name in reader.get_variable_to_shape_map():
      if old_name in vars_to_rename:
        new_name = vars_to_rename[old_name]
      else:
        new_name = old_name
      new_checkpoint_vars[new_name] = tf.Variable(reader.get_tensor(old_name))

    init = tf.global_variables_initializer()
    saver = tf.train.Saver(new_checkpoint_vars)

    with tf.Session() as sess:
      sess.run(init)
      saver.save(sess, "/home/ndscbigdata/work/change/tf/gan/im2txt/ckpt/newmodel.ckpt-2000000")
    print("checkpoint file rename successful... ")
 

然後將run.sh中的 ：CHECKPOINT_PATH、VOCAB_FILE、IMAGE_FILE等三項改成自己目錄中對應的位置
./run.sh
搞定
執行結果