1. 程式人生 > >SSD-Tensorflow測試程式碼問題

SSD-Tensorflow測試程式碼問題

本文的原始碼地址是https://github.com/balancap/SSD-Tensorflow

1.測試程式碼中讀取.ckpt檔案錯誤。

程式碼給出的checkpoints資料夾中並沒有.ckpt字尾檔案,而是一個.data檔案和.index檔案,這是由於新版saver造成的。

使用:saver = tf.train.Saver(write_version=tf.train.SaverDef.V1)

替換:saver = tf.train.Saver()